Update app.py

app.py

@@ -30,7 +30,7 @@ def predict(image):
     inputs = processor(images=image, return_tensors="pt").to(device, torch.float16)
     new_eos_token_id = 13
     with torch.no_grad():
-        generated_ids = modelA.generate(**inputs, max_length=100,
+        generated_ids = model.generate(**inputs, max_length=100,
                                       eos_token_id=new_eos_token_id)
     generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)
     return f"{generated_caption[0]}"