Update main.py

main.py

@@ -17,7 +17,7 @@ app.add_middleware(  # add the middleware
 def hello():
     return "Hello, I'm Artist"
 
-async def generate(prompt):
+def generate(prompt):
         result = Linlada._create_completion(model='gpt-4', messages=[
                                         {"role": "user", "content": prompt}], stream=True)
         chat = []                            
@@ -27,6 +27,6 @@ async def generate(prompt):
         return sentence    
 
 @app.get('/linlada/{prompt}')
-def generate_image_route(prompt: str):
-    chat = generate(prompt)        
+async def generate_image_route(prompt: str):
+    chat = await generate(prompt)        
     return chat