Update app.py

app.py

@@ -3,7 +3,7 @@ import streamlit as st
 import requests
 from PIL import Image
 import numpy as np
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoProcessor, AutoModelForPreTraining
 import torch
 from huggingface_hub import login
 
@@ -12,9 +12,8 @@ token = os.getenv("HF_Token")
 login(token)
 
 # Initialize the Hugging Face model
-model_name = "google/paligemma-3b-mix-224"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoProcessor.from_pretrained("google/paligemma-3b-mix-224")
+model = AutoModelForPreTraining.from_pretrained("google/paligemma-3b-mix-224")
 
 # Function to transcribe handwritten notes using Hugging Face model
 def transcribe_handwriting(image):