Spaces:

ukumar557
/

Chatbot

Runtime error

ukumar557 commited on Nov 28, 2024

Commit

9d00c70

verified ·

1 Parent(s): 2cd6d6c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ if device.type == "cpu":
 # Load the tokenizer and model directly
 model_name = "ruslanmv/ai-medical-model-32bit"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)

 # Load the tokenizer and model directly
 model_name = "ruslanmv/ai-medical-model-32bit"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name).to(device)