Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -136,12 +136,12 @@ def upload_image(url):
|
|
| 136 |
# raise HTTPException(status_code=500, detail=f"Failed to generate image: {str(e)}")
|
| 137 |
|
| 138 |
@app.post("/generate-image")
|
| 139 |
-
|
| 140 |
-
|
| 141 |
-
|
| 142 |
runware = Runware(api_key=RUNWARE_API_KEY)
|
| 143 |
await runware.connect()
|
| 144 |
-
|
| 145 |
request_image = IImageInference(
|
| 146 |
positivePrompt=request.prompt,
|
| 147 |
model=request.model,
|
|
@@ -149,11 +149,11 @@ def upload_image(url):
|
|
| 149 |
height=request.height,
|
| 150 |
width=request.width,
|
| 151 |
)
|
| 152 |
-
|
| 153 |
images = await runware.imageInference(requestImage=request_image)
|
| 154 |
image_urls = [image.imageURL for image in images]
|
| 155 |
print("Generated Images: ", image_urls)
|
| 156 |
-
|
| 157 |
response = {
|
| 158 |
"batch": {
|
| 159 |
"prompt": request.prompt,
|
|
|
|
| 136 |
# raise HTTPException(status_code=500, detail=f"Failed to generate image: {str(e)}")
|
| 137 |
|
| 138 |
@app.post("/generate-image")
|
| 139 |
+
async def generate_image(request: ImageRequest):
|
| 140 |
+
print("Image Generation Request", request)
|
| 141 |
+
try:
|
| 142 |
runware = Runware(api_key=RUNWARE_API_KEY)
|
| 143 |
await runware.connect()
|
| 144 |
+
|
| 145 |
request_image = IImageInference(
|
| 146 |
positivePrompt=request.prompt,
|
| 147 |
model=request.model,
|
|
|
|
| 149 |
height=request.height,
|
| 150 |
width=request.width,
|
| 151 |
)
|
| 152 |
+
|
| 153 |
images = await runware.imageInference(requestImage=request_image)
|
| 154 |
image_urls = [image.imageURL for image in images]
|
| 155 |
print("Generated Images: ", image_urls)
|
| 156 |
+
|
| 157 |
response = {
|
| 158 |
"batch": {
|
| 159 |
"prompt": request.prompt,
|