Update main.py

main.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 import asyncio
 from Linlada import Chatbot, ConversationStyle
+from photo import Gimager
 
 app = FastAPI()
 
@@ -25,6 +26,11 @@ def read_root():
 @app.get("/test/{hello}")
 def hi(hello: str):
     return {"text": hello}
+
+@app.get("/image/{image}")
+def img(image: str):
+    img = Gimager(image)
+    return img
     
 @app.get('/linlada/{prompt}')
 def generate_image_route(prompt: str):