Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from transformers import pipeline
 from PIL import Image
-import pytesseract
 import json
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
@@ -9,6 +8,9 @@ from reportlab.pdfgen import canvas
 # Load BioGPT model for recommendations
 bio_gpt = pipeline("text-generation", model="microsoft/BioGPT")
 
+# Load OCR model
+ocr_model = pipeline("image-to-text", model="microsoft/trocr-base-handwritten")
+
 # Load reference ranges from dataset.json
 def load_reference_ranges(file_path="dataset.json"):
     with open(file_path, "r") as file:
@@ -17,10 +19,11 @@ def load_reference_ranges(file_path="dataset.json"):
 
 reference_ranges = load_reference_ranges()
 
-# Extract text from uploaded image using OCR
+# Extract text from uploaded image using Hugging Face OCR
 def extract_text_from_image(image_path):
     try:
-        text = pytesseract.image_to_string(Image.open(image_path))
+        image = Image.open(image_path)
+        text = ocr_model(image)[0]["generated_text"]
         return text
     except Exception as e:
         return f"Error extracting text: {e}"
@@ -108,7 +111,7 @@ interface = gr.Interface(
     css="""
         body {
             font-family: 'Arial', sans-serif;
-            background-color: #f9f9f9;
+            background-color: #000000;
         }
         .gradio-container {
             color: #333;