Update app.py

app.py

@@ -3,6 +3,10 @@ import gradio as gr
 import numpy as np
 from transformers import pipeline
 
+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")
 
@@ -24,7 +28,7 @@ description = "Demo that compares [T5-large](https://huggingface.co/t5-large) an
 def inference(text):
   output_flan = pipe_flan(text)[0]["generated_text"]
   output_vanilla = pipe_vanilla(text)[0]["generated_text"]
-  return [output_flan, output_vanilla_
+  return [output_flan, output_vanilla]
 
 io = gr.Interface(
   inference,