Update app.py

app.py

@@ -4,12 +4,12 @@ import torch
 
 access_token = "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"
 
-#set up the model (large version of DialoGPT)
+#set up the model
 tokenizer = AutoTokenizer.from_pretrained("sunshineccl/Mental-health-chatbot" ,use_auth_token=access_token)
 
 model = AutoModelForCausalLM.from_pretrained("sunshineccl/Mental-health-chatbot" ,use_auth_token=access_token)
-#Defining a predict function
 
+#Defining a predict function
 def predict(input, history=[]):
     # tokenize the new input sentence
     new_user_input_ids = tokenizer.encode(input + tokenizer.eos_token, return_tensors='pt')