Update app.py

app.py

@@ -24,7 +24,7 @@ def generate(model_name, image, text):
     # model_pipe.to("cuda")
     print("Generating!")
     with torch.autocast(device_type='cuda', dtype=torch.bfloat16):
-        result = model_pipe(images=[image_tensor], texts=[text])
+        result = model_pipe(images=[image], texts=[text])
     return result
 
 iface = gr.Interface(