Update main.py

main.py

@@ -1,9 +1,4 @@
-from flask import url_for
-from diffusers import StableDiffusionPipeline
-import torch
-from fastapi import FastAPI, Response  
-from fastapi.middleware.cors import CORSMiddleware
-from auth_token import auth_token
+import Linlada
 
 app = FastAPI()
 
@@ -30,14 +25,16 @@ pipe.safety_checker = dummy
 def hello():
     return "Hello, I'm Artist"
 
+async def generate(prompt):
+    result = await Linlada._create_completion(model='gpt-4', messages=[
+                                     {"role": "user", "content": promp}], stream=False) # alterative model setting
+    for message in response:
+    return message
 
 @app.get("/gen/{prompt}")
 def generate_image(prompt: str):
-    image = pipe(prompt,
-                  guidance_scale=8.5  # how strict to follow the prompt
-                  ).images[0]
-    # Save the image
-    image.save('image.png')
-
-
-    return "Gen done"
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    result = loop.run_until_complete(generate(prompt))
+    loop.close()
+    return result