Update app.py

app.py

@@ -9,9 +9,9 @@ def translation(text):
 
     model_checkpoint = "bigscience/mt0-base"
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint).to('cuda')
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
     #inference
-    inputs = tokenizer("translate to arabic : hi, I am salma and you?", return_tensors="pt").to("cuda")
+    inputs = tokenizer("translate to arabic : hi, I am salma and you?", return_tensors="pt")
     output = model.generate(**inputs)
     output = tokenizer.decode(output.cpu().numpy()[0])