Update app.py

app.py

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