Update app.py

app.py

@@ -126,14 +126,17 @@ def response(instruction, input_image=None ):
     if input_image is None:
         output="Image Generation"
     else:
-        text = instruction
-        labels = ["Image Editing", "Image Generation"]
-        classification = client.zero_shot_classification(text, labels, multi_label=True)
-        output = classification[0]
-        output = str(output)
-        if "Editing" in output:
-            output = "Image Editing"
-        else:
+        try:
+            text = instruction
+            labels = ["Image Editing", "Image Generation"]
+            classification = client.zero_shot_classification(text, labels, multi_label=True)
+            output = classification[0]
+            output = str(output)
+            if "Editing" in output:
+                output = "Image Editing"
+            else:
+                output = "Image Generation"
+        except error:
             output = "Image Generation"
     return output
 
@@ -167,14 +170,13 @@ examples=[
         [
             "Image Generation",
             None,
-            "Ironman fighting with hulk, wall painting",
+            "An alien grasping a sign board contain word 'ALIEN', futuristic, neonpunk, detailed",
 
         ],
         [
             "Image Generation",
             None,
             "Beautiful Eiffel Tower at Night",
-
         ],
     ]
 
@@ -202,7 +204,7 @@ with gr.Blocks(css=css) as demo:
                 show_label=False,
                 interactive=True,
             )
-        seed = gr.Number(value=1371, precision=0, label="Seed", interactive=True)
+        seed = gr.Number(value=1371, step=1, label="Seed", interactive=True)
 
     with gr.Row():
         width =  gr.Slider( label="Width", minimum=256, maximum=2048, step=64, value=1024)