Update app.py

app.py

@@ -38,13 +38,14 @@ def save_image_to_s3(image):
     filename = f"generated_image_{int(time.time())}.png"
 
     # Upload to S3
-    s3_client.put_object(Bucket=S3_BUCKET, Key=filename, Body=img_byte_arr)
+    s3_client.put_object(Bucket=S3_BUCKET, 
+                         Key=filename, 
+                         Body=img_byte_arr, 
+                         ContentType='image/png',
+                         ACL='public-read')
 
-    # Generate a pre-signed URL (valid for 1 hour)
-    url = s3_client.generate_presigned_url('get_object',
-                                           Params={'Bucket': S3_BUCKET,
-                                                   'Key': filename},
-                                           ExpiresIn=3600)
+    # Generate a public URL
+    url = f"https://{S3_BUCKET}.s3.{S3_REGION}.amazonaws.com/{filename}"
     return url
 
 def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=5.0, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):