Update app.py

app.py

@@ -33,12 +33,15 @@ def answer_question(image, question, model, processor):
     
     
     image = Image.open(image)
-    
-
 
     inputs = processor(image, question, return_tensors="pt").to("cuda", torch.float16)
 
-    out = model.generate(**inputs, max_length=200, min_length=20)
+    if isinstance(model, torch.nn.DataParallel):
+    # Use the 'module' attribute to access the original model
+        out = model.module.generate(**inputs)
+    else:
+       
+        out = model.generate(**inputs, max_length=200, min_length=20)
 
     answer = processor.decode(out[0], skip_special_tokens=True).strip()
     return answer