Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -131,6 +131,7 @@ model_m = Qwen2VLForConditionalGeneration.from_pretrained(
|
|
| 131 |
torch_dtype=torch.float16
|
| 132 |
).to("cuda").eval()
|
| 133 |
|
|
|
|
| 134 |
def generate_qwen_ocr(input_text: str, image):
|
| 135 |
"""
|
| 136 |
Uses the Qwen2VL OCR model to process an image along with text.
|
|
|
|
| 131 |
torch_dtype=torch.float16
|
| 132 |
).to("cuda").eval()
|
| 133 |
|
| 134 |
+
@spaces.GPU
|
| 135 |
def generate_qwen_ocr(input_text: str, image):
|
| 136 |
"""
|
| 137 |
Uses the Qwen2VL OCR model to process an image along with text.
|