Spaces:
Runtime error
Runtime error
mynkchaudhry
commited on
Commit
•
df00160
1
Parent(s):
e25c164
upload all the files
Browse files
app.py
CHANGED
@@ -4,8 +4,8 @@ from PIL import Image
|
|
4 |
import torch
|
5 |
|
6 |
# Load model and processor with trust_remote_code=True
|
7 |
-
model = AutoModelForCausalLM.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA",
|
8 |
-
processor = AutoProcessor.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA",
|
9 |
|
10 |
def generate_response(image, question):
|
11 |
try:
|
|
|
4 |
import torch
|
5 |
|
6 |
# Load model and processor with trust_remote_code=True
|
7 |
+
model = AutoModelForCausalLM.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA", trust_remote_code=True)
|
8 |
+
processor = AutoProcessor.from_pretrained("mynkchaudhry/Florence-2-FT-DocVQA", trust_remote_code=True)
|
9 |
|
10 |
def generate_response(image, question):
|
11 |
try:
|