Update app.py

app.py

@@ -7,8 +7,8 @@ import torch
 
 # Load the model and processor
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = AutoModelForCausalLM.from_pretrained("sahilnishad/Florence-2-FT-DocVQA").to(device)
-processor = AutoProcessor.from_pretrained("sahilnishad/Florence-2-FT-DocVQA")
+model = AutoModelForCausalLM.from_pretrained("sahilnishad/Florence-2-FT-DocVQA", trust_remote_code=True).to(device)
+processor = AutoProcessor.from_pretrained("sahilnishad/Florence-2-FT-DocVQA", trust_remote_code=True)
 
 # Function to run inference
 def get_answer(task_prompt, question, image):