Upload folder using huggingface_hub

api/index.py

@@ -21,11 +21,8 @@ load_dotenv(dotenv_path)
 OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
 REPLICATE_API_TOKEN = os.getenv('REPLICATE_API_TOKEN')
 
-
-
 client = OpenAI()
 
-
 def call_openai(pil_image):
     # Save the PIL image to a bytes buffer
     buffered = io.BytesIO()
@@ -132,5 +129,6 @@ def image_classifier(moodboard, starter_image, image_strength, prompt):
 
     return images
 
-demo = gr.Interface(fn=image_classifier, inputs=["image", "image", gr.Slider(0, 1, step=0.025, value=0.2, label="Image Strength"), "text"], outputs=["image", "image", "image"])
-demo.launch(share=False)
+header = "Set up APIs on HuggingFace or use free at https://app.idai.tools/"
+demo = gr.Interface(fn=image_classifier, inputs=["image", "image", gr.Slider(0, 1, step=0.05, value=0.2, label="Image Strength"), "text"], outputs=["image", "image", "image"], title=header)
+demo.launch(share=False)