Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,13 +60,6 @@ model_x = Qwen2VLForConditionalGeneration.from_pretrained(
|
|
| 60 |
).to(device).eval()
|
| 61 |
|
| 62 |
# Load Aya-Vision-8b
|
| 63 |
-
MODEL_ID_A = "CohereForAI/aya-vision-8b"
|
| 64 |
-
processor_a = AutoProcessor.from_pretrained(MODEL_ID_A, trust_remote_code=True)
|
| 65 |
-
model_a = AutoModelForImageTextToText.from_pretrained(
|
| 66 |
-
MODEL_ID_A,
|
| 67 |
-
trust_remote_code=True,
|
| 68 |
-
torch_dtype=torch.float16
|
| 69 |
-
).to(device).eval()
|
| 70 |
|
| 71 |
# Load olmOCR-7B-0725
|
| 72 |
MODEL_ID_W = "allenai/olmOCR-7B-0725"
|
|
@@ -127,9 +120,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
|
|
| 127 |
elif model_name == "Nanonets-OCR-s":
|
| 128 |
processor = processor_v
|
| 129 |
model = model_v
|
| 130 |
-
|
| 131 |
-
processor = processor_a
|
| 132 |
-
model = model_a
|
| 133 |
elif model_name == "olmOCR-7B-0725":
|
| 134 |
processor = processor_w
|
| 135 |
model = model_w
|
|
|
|
| 60 |
).to(device).eval()
|
| 61 |
|
| 62 |
# Load Aya-Vision-8b
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
|
| 64 |
# Load olmOCR-7B-0725
|
| 65 |
MODEL_ID_W = "allenai/olmOCR-7B-0725"
|
|
|
|
| 120 |
elif model_name == "Nanonets-OCR-s":
|
| 121 |
processor = processor_v
|
| 122 |
model = model_v
|
| 123 |
+
|
|
|
|
|
|
|
| 124 |
elif model_name == "olmOCR-7B-0725":
|
| 125 |
processor = processor_w
|
| 126 |
model = model_w
|