Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 from transformers import pipeline
 
 # Model ID for Llama 3 8B instruct (replace with the exact model you want)
-MODEL_ID = "tiiuae/falcon-40b"
+MODEL_ID = "rumb/nano-mistral"
 
 # Load the text-generation pipeline with device_map="auto" to use GPU if available
 generator = pipeline(