Update app.py

app.py

@@ -50,8 +50,16 @@ def gradio_gen_process(img,prompt,ps=0.6,style_model=None,ls=0.6,jp=None,recipie
 
   #Image gen 
   print(img)
-  img,file_saveloc=Gen_image(prompt,img=img,ps=ps,lr=ls,model=model,save_loc=save_loc) 
-  
+  try:
+      img,file_saveloc=Gen_image(prompt,img=img,ps=ps,lr=ls,model=model,save_loc=save_loc) 
+      gallery.append(img)
+  except:
+      recipient_email=None
+      print("Error:Image gen fail")
+      img=None
+      pass
+
+      
   #Send email
   if recipient_email:
     send_email(file_saveloc,recipient_email,prompt=prompt)
@@ -59,7 +67,8 @@ def gradio_gen_process(img,prompt,ps=0.6,style_model=None,ls=0.6,jp=None,recipie
   else:
     print("Email not sent")
 
-  gallery.append(img)
+  
+  
   return img,gallery