Update model.py

model.py

@@ -50,7 +50,7 @@ def run(message: str,
         top_k=top_k,
         temperature=temperature,
     )
-    stream = client.generate_stream(prompt, **generate_kwargs)
+    stream = client.text_generation(prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
     for response in stream:
         if any([end_token in response.token.text for end_token in [EOS_STRING, EOT_STRING]]):