Update app.py

app.py

@@ -1,15 +1,16 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import AutoProcessor, AutoModelForPreTraining 
 import pytesseract
 import cv2
 import numpy as np
 from PIL import Image
 from huggingface_hub import login
 
-login()
+login(HF_token)
 
 # Load the model
-model = pipeline("text2text-generation", model="google/paligemma-3b-mix-224")
+processor = AutoProcessor.from_pretrained("google/paligemma-3b-mix-224")
+model = AutoModelForPreTraining.from_pretrained("google/paligemma-3b-mix-224")
 
 # Function to extract text from image using OCR
 def extract_text_from_image(image_file):