prithivMLmods commited on
Commit
9a91d7d
·
verified ·
1 Parent(s): f21867a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -102,10 +102,10 @@ def generate_image(model_name: str, text: str, image: Image.Image,
102
  elif model_name == "DREX-062225-7B-exp":
103
  processor = processor_x
104
  model = model_x
105
- elif model_name == "olmOCR-7B-0225":
106
  processor = processor_o
107
  model = model_o
108
- elif model_name == "typhoon-ocr-3b":
109
  processor = processor_t
110
  model = model_t
111
  else:
@@ -158,10 +158,10 @@ def generate_video(model_name: str, text: str, video_path: str,
158
  elif model_name == "DREX-062225-7B-exp":
159
  processor = processor_x
160
  model = model_x
161
- elif model_name == "olmOCR-7B-0225":
162
  processor = processor_o
163
  model = model_o
164
- elif model_name == "typhoon-ocr-3b":
165
  processor = processor_t
166
  model = model_t
167
  else:
 
102
  elif model_name == "DREX-062225-7B-exp":
103
  processor = processor_x
104
  model = model_x
105
+ elif model_name == "olmOCR-7B-0225-preview":
106
  processor = processor_o
107
  model = model_o
108
+ elif model_name == "Typhoon-OCR-3B":
109
  processor = processor_t
110
  model = model_t
111
  else:
 
158
  elif model_name == "DREX-062225-7B-exp":
159
  processor = processor_x
160
  model = model_x
161
+ elif model_name == "olmOCR-7B-0225-preview":
162
  processor = processor_o
163
  model = model_o
164
+ elif model_name == "Typhoon-OCR-3B":
165
  processor = processor_t
166
  model = model_t
167
  else: