Spaces:

JaiPatel4717
/

NutritionChatBot

Sleeping

JaiPatel4717 commited on 7 days ago

Commit

e4e473d

•

1 Parent(s): dc59162

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 # Load the model and tokenizer
 model_name = "Tom158/Nutri_Assist"
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 # Set pad token if not already set

 # Load the model and tokenizer
 model_name = "Tom158/Nutri_Assist"
+model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 # Set pad token if not already set