Update app.py

app.py

@@ -279,8 +279,6 @@ def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p
     print(buffer)
     print("---------")
 
-    return PlainTextResponse(buffer)
-
     # with torch.no_grad():
     #     thread = Thread(target=model.generate, kwargs=gen_kwargs)
     #     thread.start()
@@ -296,6 +294,10 @@ def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p
 
     # return PlainTextResponse(buffer)
 
+return PlainTextResponse(buffer)
+
+
+