Update app.py

app.py

@@ -4,8 +4,8 @@ import torch
 
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
-model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
+tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-small')
+model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-small')
 
 def predict(input, history=[]):
     # tokenize the new input sentence
@@ -25,7 +25,7 @@ def predict(input, history=[]):
 
 
 gr.Interface(fn=predict,
-    title="DialoGPT-medium",
+    title="DialoGPT-small",
     inputs=["text", "state"],
     outputs=["chatbot", "state"],
     allow_screenshot=False,