fix

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from lavis.models import load_model_and_preprocess
 from PIL import Image
+import torch
 
 def process(input_image, prompt):
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')