Update app.py

app.py

@@ -8,7 +8,11 @@ from threading import Thread
 
 # Loading the tokenizer and model from Hugging Face's model hub.
 tokenizer = AutoTokenizer.from_pretrained("soketlabs/pragna-1b", token=os.environ.get('HF_TOKEN'))
-model = AutoModelForCausalLM.from_pretrained("soketlabs/pragna-1b", token=os.environ.get('HF_TOKEN'))
+model = AutoModelForCausalLM.from_pretrained(
+    "soketlabs/pragna-1b", 
+    token=os.environ.get('HF_TOKEN'),
+    revision='3c5b8b1309f7d89710331ba2f164570608af0de7'
+)
 model.load_adapter('soketlabs/pragna-1b-it-v0.1', token=os.environ.get('HF_TOKEN'))
 
 # using CUDA for an optimal experience