fjf

app.py

@@ -114,7 +114,7 @@ print("Model trained and saved successfully.")
 print(f"Dataset integrated. Total rows: {len(combined_ds)}")
 
 # --- 2. MODEL LOADING ---
-MODEL_ID = "abdelac/tinyllama"
+MODEL_ID = "abdelac/Mistral_Test"
 
 # We assume these are defined globally for the generate_text function
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)