4th

app.py

@@ -8,7 +8,7 @@ device = 'cpu'
 
 class Generator:
     def __init__(self) -> None:
-        self.generator = pipeline('text-generation', model="EleutherAI/gpt-neo-1.3B", do_sample=True)
+        self.generator = pipeline('text-generation', model="EleutherAI/gpt-neo-125M", do_sample=True)
         self.history = "Human: Can you tell me the weather forecast for tomorrow?\nAssistant: Try checking a weather app like a normal person.\nHuman: Can you help me find a good restaurant in the area\nAssistant: Try asking someone with a functioning sense of taste.\n"
 
     def generate(self, text, max_len=400, temp=0.8):