Update app.py

app.py

@@ -158,6 +158,10 @@ for model_name in model_list:
 
 
 def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
+
+    if width == 0 or height == 0:
+        raise gr.Error("width&height should be > 0")
+    
     result = prodia_client.generate({
         "prompt": prompt,
         "negative_prompt": negative_prompt,
@@ -181,6 +185,9 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
 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")
+
+    if width == 0 or height == 0:
+        raise gr.Error("width&height should be > 0")
         
     result = prodia_client.transform({
         "imageData": image_to_base64(input_image),