Update main.py

main.py

@@ -9,7 +9,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 
 app = FastAPI()
-name = "microsoft/DialoGPT-small"
+name = "PygmalionAI/pygmalion-1.3b"
 
 # microsoft/DialoGPT-small
 # microsoft/DialoGPT-medium
@@ -48,6 +48,6 @@ def read_root(data: req):
   generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
   
   answer_data = { "answer": generated_text }
-  print("Answered with:", answer_data)
+  print("Answer:", generated_text)
   
   return answer_data