Update app.py

app.py

@@ -41,7 +41,7 @@ def query_image(prompt):
   payload={"inputs":prompt}
   for attempt in range(max_retries):
     response=requests.post(image_API,headers=headers,json=payload)
-    if response.status_code==200 and response.get.headers('Content-Type','').startswith('image/'):
+    if response.status_code == 200 and response.headers.get('Content-Type', '').startswith('image/'):
       image_path='generated_image.png'
       with open(image_path,'wb') as f:
         f.write(response.content)