sharktide commited on
Commit
49bc6b1
·
verified ·
1 Parent(s): 2ccd595

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -68,12 +68,14 @@ async def generate_image(prompt: str, request: Request):
68
  detail=f"Pollinations error: {response.status_code}"
69
  )
70
 
 
71
  return Response(
72
  content=response.content,
73
- media_type=response.headers.get("content-type", "image/png")
74
  )
75
 
76
 
 
77
  # -----------------------------
78
  # EXISTING MODELS SCRAPER
79
  # -----------------------------
 
68
  detail=f"Pollinations error: {response.status_code}"
69
  )
70
 
71
+ # Pollinations always returns JPEG
72
  return Response(
73
  content=response.content,
74
+ media_type="image/jpeg"
75
  )
76
 
77
 
78
+
79
  # -----------------------------
80
  # EXISTING MODELS SCRAPER
81
  # -----------------------------