Change model

app.py

@@ -32,7 +32,7 @@ def initialize_model():
         # Access token might be needed for some models
         # token = huggingface_hub.get_token()
         
-        model_name = "Daemontatox/DocumentCogito"
+        model_name = "microsoft/phi-2"
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForCausalLM.from_pretrained(
             model_name,
@@ -40,7 +40,7 @@ def initialize_model():
             device_map="auto",
             trust_remote_code=True
         )
-        logger.info("Successfully initialized DocumentCogito model and tokenizer.")
+        logger.info("Successfully initialized phi-2 model and tokenizer.")
         return model, tokenizer
     except Exception as e:
         logger.exception("Error initializing Hugging Face model.")
@@ -54,7 +54,7 @@ model, tokenizer = initialize_model()
 ###############################################################################
 def generate_with_model(prompt, max_length=1400, temperature=0.3):
     """Generate text using the Hugging Face model."""
-    logger.info("Generating text with DocumentCogito model.")
+    logger.info("Generating text with phi-2 model.")
     
     try:
         inputs = tokenizer(prompt, return_tensors="pt").to(model.device)