Update app.py

app.py

@@ -171,10 +171,16 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
 
     job = prodia_client.wait(result)
 
+    if job['status'] != "succeeded":
+        raise gr.Error("job failed")
+
     return job["imageUrl"]
 
 
 def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
+    if input_image is None:
+        raise gr.Error("Please add an image to run img2img")
+        
     result = prodia_client.transform({
         "imageData": image_to_base64(input_image),
         "denoising_strength": denoising,
@@ -191,6 +197,9 @@ def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampl
 
     job = prodia_client.wait(result)
 
+    if job['status'] != "succeeded":
+        raise gr.Error("job failed")
+
     return job["imageUrl"]