Update app.py

app.py

@@ -1,6 +1,4 @@
 import gradio as gr
-import random
-import os
 
 
 models = {
@@ -17,16 +15,25 @@ default_negative_prompt = (
     "Cartoonish appearance, Plastic look, Grainy, Unnatural expressions, Crossed eyes, Mutated clothing, "
     "Artifacts, Uncanny valley, Doll-like features, Bad symmetry, Uneven skin tones, Extra teeth, "
     "Unrealistic hair texture, Dark shadows on face, Overexposed face, Cluttered background, Text, watermark, "
-    "or signature, Over-processed"
+    "or signature, Over-processed, Low quality, Blurry, Low resolution, Pixelated, Oversaturated, Too dark, Overexposed, Poor lighting, "
+    "Unclear, Text or watermark, Distorted faces, Extra limbs or fingers, Disfigured, Grainy, Overly stylized, Cartoonish, "
+    "Unrealistic anatomy, Bad proportions, Unrealistic colors, Artificial look, Background noise, Unwanted objects, Repetitive patterns, Artifacting, Abstract shapes, Out of focus"
 )
 
+
 def generate_image(text, negative_prompt):
     result_images = {}
     for model_name, model in models.items():
-        result_images[model_name] = model(text, negative_prompt=negative_prompt) 
+        try:
+
+            result_images[model_name] = model(text, negative_prompt=negative_prompt)
+        except TypeError:
+
+            result_images[model_name] = model(text)
     
     return [result_images[model_name] for model_name in models]
 
+
 interface = gr.Interface(
     fn=generate_image,
     inputs=[
@@ -38,4 +45,5 @@ interface = gr.Interface(
     description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
 )
 
-interface.launch()
+
+interface.launch()