Upload folder using huggingface_hub

api/index.py

@@ -68,6 +68,10 @@ def call_openai(pil_image):
         raise gr.Error("Unknown Error")
 
 
+# Todo -- better prompt generator, add another LLM layer combining the user prompt and moodboard description (in the case of the jacket, 'high quality render of yellow jacket, its fabric is a pattern of cosmic etc etc' worked well)
+# Could even do this 4 different times to get more diversity of renders
+# Add "simple" to prompt before word
+
 def image_classifier(moodboard, prompt):
 
     if moodboard is not None:
@@ -82,7 +86,7 @@ def image_classifier(moodboard, prompt):
 
     input = {
         "prompt": "high quality render of " + prompt + ", " + openai_response[12:],
-        "negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch",
+        "negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch, logo, buttons",
         "output_format": "jpg"
     }
     
@@ -120,7 +124,7 @@ def image_classifier(moodboard, prompt):
         "width": 768,
         "height": 768,
         "prompt": "high quality render of " + prompt + ", " + openai_response[12:],
-        "negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch",
+        "negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch, logo, buttons",
         "refine": "expert_ensemble_refiner",
         "apply_watermark": False,
         "num_inference_steps": 25,