Update app.py

app.py

@@ -16,7 +16,9 @@ print("loading model")
 repo_name = "nvidia/Hymba-1.5B-Instruct"
 
 tokenizer = AutoTokenizer.from_pretrained(repo_name, trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained(repo_name, trust_remote_code=True)
+#model = AutoModelForCausalLM.from_pretrained(repo_name, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolLM2-1.7B-Instruct", trust_remote_code=True)
+
 model = model.cuda().to(torch.bfloat16)
 
 print("model is loaded")