Update app.py

app.py

@@ -109,15 +109,20 @@ def infer(
     num_inference_steps=50,
     progress=gr.Progress(track_tqdm=True),
 ):
+
+    print(f"Generating for prompt: \n\t{prompt}\n\t{seed}\n\t{aspect_ratio}\n\t{num_inference_steps}")
   
     seed = sanitize_seed(seed)
     
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
-        
-    width, height = aspect_ratios[aspect_ratio]
+
+    try:
+        width, height = aspect_ratios[aspect_ratio]
+    except:
+        width, height = (640, 480)
+    
     
-    print(f"Generating for prompt: \n\t{prompt}\n\t{seed}\n\t{aspect_ratio}\n\t{num_inference_steps}")
     
     image = pipe(
         prompt=prompt,