Electro0023 commited on
Commit
265d864
·
verified ·
1 Parent(s): 697cc2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -3
app.py CHANGED
@@ -1,12 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import torch
2
  import gradio as gr
3
  from transformers import AutoProcessor, AutoModelForVision2Seq
4
  from PIL import Image
5
  import os
6
 
7
- os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
8
 
9
- model_id = "HuggingFaceTB/SmolVLM-256M-Instruct" # smallest possible - 256M params
10
 
11
  print("Loading processor...")
12
  processor = AutoProcessor.from_pretrained(model_id)
@@ -67,7 +81,7 @@ demo = gr.Interface(
67
  inputs=gr.Image(type="pil", label="Upload NEET Question Image"),
68
  outputs=gr.Textbox(label="Extracted Text", lines=20),
69
  title="NEET Question Extractor",
70
- description="Upload a scanned NEET question paper image to extract text"
71
  )
72
 
73
  demo.launch()
 
1
+ # Force install FIRST before any other imports
2
+ import subprocess
3
+ import sys
4
+
5
+ print("Installing correct transformers version...")
6
+ subprocess.run([
7
+ sys.executable, "-m", "pip", "install",
8
+ "--upgrade", "--force-reinstall",
9
+ "--no-deps",
10
+ "transformers==4.52.4"
11
+ ], check=True)
12
+ print("Done installing transformers")
13
+
14
+ # Now safe to import
15
  import torch
16
  import gradio as gr
17
  from transformers import AutoProcessor, AutoModelForVision2Seq
18
  from PIL import Image
19
  import os
20
 
21
+ print(f"Transformers version: {__import__('transformers').__version__}")
22
 
23
+ model_id = "HuggingFaceTB/SmolVLM-256M-Instruct"
24
 
25
  print("Loading processor...")
26
  processor = AutoProcessor.from_pretrained(model_id)
 
81
  inputs=gr.Image(type="pil", label="Upload NEET Question Image"),
82
  outputs=gr.Textbox(label="Extracted Text", lines=20),
83
  title="NEET Question Extractor",
84
+ description="Upload scanned NEET question paper image to extract text"
85
  )
86
 
87
  demo.launch()