Update app.py

app.py

@@ -5,7 +5,7 @@ from gpt import get_model
 model_small, tokenizer_small = get_model("gpt2")
 
 def predict(inp, model_type):
-    if model_type == "gpt2":
+    if model_type == "gpt2-small":
         model, tokenizer = model_small, tokenizer_small
         input_ids = tokenizer.encode("user:"+inp+"\nai:", return_tensors='tf')
         beam_output = model.generate(input_ids, max_length=180, num_beams=5,