Update app.py

app.py

@@ -22,8 +22,8 @@ article = "<p style='text-align: center'><a href='https://medium.com/geekculture
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-#tokenizer = AutoTokenizer.from_pretrained("kingabzpro/DialoGPT-small-Rick-Bot")
-#model = AutoModelForCausalLM.from_pretrained("kingabzpro/DialoGPT-small-Rick-Bot")
+tokenizer = AutoTokenizer.from_pretrained("kingabzpro/DialoGPT-small-Rick-Bot")
+model = AutoModelForCausalLM.from_pretrained("kingabzpro/DialoGPT-small-Rick-Bot")
 
 def predict(input, history=[]):
     # tokenize the new input sentence
@@ -40,7 +40,7 @@ def predict(input, history=[]):
     
     return response, history
 
-gr.Interface.load("huggingface/kingabzpro/DialoGPT-small-Rick-Bot",inputs = "textbox", outputs = "chatbot",allow_flagging = "manual",theme ="grass",title = title,lagging_callback=hf_writer,description = description, article = article ).launch(enable_queue=True)  # customizes the input component
+gr.Interface(fn = predict,inputs = "textbox", outputs = "chatbot",allow_flagging = "manual",theme ="grass",title = title,lagging_callback=hf_writer,description = description, article = article ).launch(enable_queue=True)  # customizes the input component
 
  #theme ="grass", 
  #title = title,