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.
|