Update app.py

app.py

@@ -1,5 +1,7 @@
 
+import torch
 from transformers import pipeline
+import gradio as gr
 
 # Initialize the pipeline
 pipe = pipeline(