Update app.py

app.py

@@ -7,13 +7,12 @@ import torch
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 
-pipe_flan = pipeline("text2text-generation", model="google/flan-t5-large", device="cuda")
-pipe_vanilla = pipeline("text2text-generation", model="t5-large", device="cuda")
+pipe_flan = pipeline("text2text-generation", model="google/flan-t5-large", device="cuda:0")
+pipe_vanilla = pipeline("text2text-generation", model="t5-large", device="cuda:0")
 
 examples = [
-  ["Translatio:0n"],
   ["Please answer to the following question. Who is going to be the next Ballon d'or?"],
-  ["Q: Can Geo:0ffrey Hinton have a conversation with George Washington? Give the rationale before answering."],
+  ["Q: Can Geoffrey Hinton have a conversation with George Washington? Give the rationale before answering."],
   ["Please answer the following question. What is the boiling point of Nitrogen?"],
   ["Answer the following yes/no question. Can you write a whole Haiku in a single tweet?"],
   ["Answer the following yes/no question by reasoning step-by-step. Can you write a whole Haiku in a single tweet?"],