Spaces:

mynkchaudhry
/

Image2text

Runtime error

mynkchaudhry commited on Jul 16, 2024

Commit

5b7cddd

1 Parent(s): df00160

upload all the files

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,9 +3,9 @@ from transformers import AutoModelForCausalLM, AutoProcessor
 from PIL import Image
 import torch
-# Load model and processor with trust_remote_code=True
 model = AutoModelForCausalLM.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA", trust_remote_code=True)
-processor = AutoProcessor.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA", trust_remote_code=True)
 def generate_response(image, question):
     try:

 from PIL import Image
 import torch
+# Load model and processor
 model = AutoModelForCausalLM.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA", trust_remote_code=True)
+processor = AutoProcessor.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA")
 def generate_response(image, question):
     try: