Upload models

app.py

@@ -96,15 +96,15 @@ GENERATOR_MODEL_NAME = "Ah1111/Generator_Model"
 # Load generator model (T5)
 @st.cache_resource
 def load_llm():
-    tokenizer = T5Tokenizer.from_pretrained(GENERATOR_MODEL_NAME)
-    model = T5ForConditionalGeneration.from_pretrained(GENERATOR_MODEL_NAME)
+    tokenizer = T5Tokenizer.from_pretrained("Ah1111/Generator_Model")
+    model = T5ForConditionalGeneration.from_pretrained("Ah1111/Generator_Model")
     return tokenizer, model
 
 # Load embedding model (custom Hugging Face model)
 @st.cache_resource
 def load_embedding_model():
-    tokenizer = AutoTokenizer.from_pretrained(EMBEDDING_MODEL_NAME)
-    model = AutoModel.from_pretrained(EMBEDDING_MODEL_NAME)
+    tokenizer = AutoTokenizer.from_pretrained("Ah1111/Embedding_Model")
+    model = AutoModel.from_pretrained("Ah1111/Embedding_Model")
     return tokenizer, model
 
 # Load FAISS index and texts