rahul7star commited on
Commit
551b2a9
·
verified ·
1 Parent(s): b3490eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -10
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
- elif model_name == "Aya-Vision-8B":
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