Update app.py

app.py

@@ -4,6 +4,9 @@ import io
 import random
 import os
 from PIL import Image
+from huggingface_hub import HfApi
+
+login(token=os.getenv("HF_READ_TOKEN"))
 
 API_URL = "https://api-inference.huggingface.co/models/openskyml/open-diffusion-v1"
 API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
@@ -48,7 +51,17 @@ def query(prompt, is_negative=False, image_style="None style", steps=8, cfg_scal
 
         image_bytes = requests.post(API_URL, headers=headers, json=payload).content
         image = Image.open(io.BytesIO(image_bytes))
+
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj=image,
+            path_in_repo="images/img_"+random.randint(1, 2147483647)+".png",
+            repo_id="openskyml/prompts",
+            repo_type="dataset",
+        )
+        
         images.append(image)
+)
     return images
 
 
@@ -206,6 +219,6 @@ with gr.Blocks(css=css) as demo:
 
         
         
-    text_button.click(query, inputs=[text_prompt, negative_prompt, image_style], outputs=gallery_output)
+    text_button.click(query, inputs=[text_prompt, negative_prompt], outputs=gallery_output)
 
 demo.launch(show_api=False)