CR7CAD commited on
Commit
0755951
·
verified ·
1 Parent(s): 6ba2a3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from pdf2image import convert_from_bytes
13
  # Load the OCR Pipeline (Uses Torch)
14
  #####################################
15
  try:
16
- ocr_pipeline = pipeline("image-to-text", model="opencsg/csg-wukong-1B-VL-v0.1")
17
  st.write("Model loaded successfully!")
18
  except Exception as e:
19
  st.error(f"Error loading model: {e}")
 
13
  # Load the OCR Pipeline (Uses Torch)
14
  #####################################
15
  try:
16
+ ocr_pipeline = pipeline("image-to-text", model="YouLiXiya/tinyllava-v1.0-1.1b-hf")
17
  st.write("Model loaded successfully!")
18
  except Exception as e:
19
  st.error(f"Error loading model: {e}")