Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import re
-import torch  # Explicit import if you plan to use torch methods directly
+import torch  # Explicitly imported if you want to use torch directly
 import tempfile
 from io import BytesIO
 
@@ -13,6 +13,7 @@ from pdf2image import convert_from_bytes
 # Load the OCR Pipeline (Uses Torch)
 #####################################
 try:
+    # Make sure that you're using an updated version of the transformers library (>=4.x)
     ocr_pipeline = pipeline("image-to-text", model="YouLiXiya/tinyllava-v1.0-1.1b-hf")
     st.write("Model loaded successfully!")
 except Exception as e:
@@ -144,7 +145,7 @@ def process_resume(file_obj, company_requirements):
 #####################################
 st.title("Resume Extraction and Candidate Matching")
 st.markdown("""
-This app uses an image-to-text pipeline (powered by `alibaba-damo/mgp-str-base` and PyTorch) to 
+This app uses an image-to-text pipeline (powered by `YouLiXiya/tinyllava-v1.0-1.1b-hf` and PyTorch) to 
 extract details from uploaded resume files (PDF or image formats). It then parses critical candidate 
 information and compares it against company requirements.
 """)