Update main.py

main.py

@@ -56,17 +56,17 @@ async def searches(request: Request):
     response = agent_chain.run(input=prompt)
     return response
 
-@app.post("/imagen")
-async def generate_image(request: Request):
-    data = await request.json()
-    prompt = data['prompt']
-    model = data.get["model"]
-    sampler = data.get["sampler"]
-    seed = int(data.get["seed"])
-    neg = data.get["neg"]
-
-    response = generate_image_prodia(prompt, model, sampler, seed, neg)
-    return jsonify({"image": response})
+# @app.post("/imagen")
+# async def generate_image(request: Request):
+#     data = await request.json()
+#     prompt = data['prompt']
+#     model = data.get["model"]
+#     sampler = data.get["sampler"]
+#     seed = int(data.get["seed"])
+#     neg = data.get["neg"]
+
+#     response = generate_image_prodia(prompt, model, sampler, seed, neg)
+#     return jsonify({"image": response})
 
 class User(BaseModel):
     prompt: str