Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Use a smaller version of DialoGPT for faster inference
-model_name = "microsoft/DialoGPT-small"
+# Load the satvikag/chatbot model and tokenizer from Hugging Face
+model_name = "satvikag/chatbot"
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
@@ -18,7 +18,7 @@ def chatbot_response(user_input):
     new_user_input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors='pt')
 
     # Get the response from the model
-    bot_output = model.generate(new_user_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
+    bot_output = model.generate(new_user_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id, no_repeat_ngram_size=2)
 
     # Decode the response and return
     bot_output_text = tokenizer.decode(bot_output[:, new_user_input_ids.shape[-1]:][0], skip_special_tokens=True)