Update main.py

main.py

@@ -28,8 +28,10 @@ async def generate(prompt):
 
 @app.get('/linlada/{prompt}')
 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
+    result = Linlada._create_completion(model='gpt-4', messages=[
+                                        {"role": "user", "content": prompt}], stream=True)
+        chat = []                            
+        for message in result:
+            chat.append(message)
+        sentence = ''.join(chat)        
+        return sentence