Update app.py

app.py

@@ -61,7 +61,7 @@ def generate_text(prompt, max_length=100, temperature=0.7, top_p=0.9, top_k=50):
 
 # Load the model
 print("Loading model...")
-checkpoint_path = "checkpoints/smol-lm2-final.ckpt"
+checkpoint_path = "smol-lm2-final.ckpt"
 if not os.path.exists(checkpoint_path):
     print(f"Warning: Checkpoint not found at {checkpoint_path}")
     print("Please train the model first or specify correct checkpoint path")