Update main.py

main.py

@@ -17,16 +17,17 @@ app.add_middleware(  # add the middleware
 def hello():
     return "Hello, I'm Artist"
 
-def generate(prompt):
-        result = Linlada._create_completion(model='gpt-4', messages=[
+async def generate(prompt):
+        response = await Linlada._create_completion(model='gpt-4', messages=[
                                         {"role": "user", "content": prompt}], stream=True)
-        chat = []                            
-        for message in result:
-            chat.append(message)
+        chat = list(response)
         sentence = ''.join(chat)        
         return sentence    
 
 @app.get('/linlada/{prompt}')
-async def generate_image_route(prompt: str):
-    chat = await generate(prompt)        
-    return chat
+def generate_image_route(prompt: str):
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    result = loop.run_until_complete(generate(prompt))
+    loop.close()
+    return result