Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ def demo_process(img, question):
|
|
19 |
processor = AutoProcessor.from_pretrained(model_checkpoint)
|
20 |
model = AutoModelForDocumentQuestionAnswering.from_pretrained(model_checkpoint)
|
21 |
with torch.no_grad():
|
22 |
-
encoding = processor(img
|
23 |
outputs = model(**encoding)
|
24 |
start_logits = outputs.start_logits
|
25 |
end_logits = outputs.end_logits
|
|
|
19 |
processor = AutoProcessor.from_pretrained(model_checkpoint)
|
20 |
model = AutoModelForDocumentQuestionAnswering.from_pretrained(model_checkpoint)
|
21 |
with torch.no_grad():
|
22 |
+
encoding = processor(img, question, return_tensors="pt")
|
23 |
outputs = model(**encoding)
|
24 |
start_logits = outputs.start_logits
|
25 |
end_logits = outputs.end_logits
|