Update main.py

main.py

@@ -28,10 +28,5 @@ async def generate(prompt):
 
 @app.get('/linlada/{prompt}')
 def generate_image_route(prompt: str):
-    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
+    chat = generate(prompt)        
+    return chat