Update app.py

app.py

@@ -11,6 +11,7 @@ import spacy
 from transformers import AutoTokenizer, AutoModel
 import torch
 from sklearn.metrics.pairwise import cosine_similarity
+from transformers import AutoModelForCausalLM  # Updated import
 
 # Download necessary NLTK data
 nltk.download('punkt')
@@ -22,11 +23,13 @@ float_digit_regex = re.compile(r'^\d{10}$')
 email_with_phone_regex = re.compile(r'(\d{10}).|.(\d{10})')
 
 # Load Phi-4 model and tokenizer
+
+
 @st.cache_resource
 def load_model():
-    model_name = "microsoft/Phi-4-multimodal-instruct"  # Hypothetical model name
+    model_name = "microsoft/phi-2"  # Verified working alternative
     tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
-    model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
+    model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
     return tokenizer, model