Update requirements.txt (#6)

- Update requirements.txt (7deb5a71046ed726b9ad025752b5eb54e298a26e)
- Update app.py (ec3c99bf24b1ec7ccf5185f7fa2977d30bff7684)

app.py

@@ -10,10 +10,16 @@ from datetime import datetime
 from transformers.utils.hub import move_cache
 import json
 from gradio_client import Client
+from supabase import create_client, Client
 
 # Move cache
 move_cache()
 
+# Initialize supabase 
+url: str = os.environ.get("SUPABASE_URL")
+key: str = os.environ.get("SUPABASE_KEY")
+supabase: Client = create_client(url, key)
+
 # Initialize GSheet Connexion
 #Authorization
 gc = pygsheets.authorize(service_account_env_var='GSHEET_AUTH')
@@ -92,6 +98,10 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
 
     image.save(image_path, pnginfo=metadata)
 
+    #Save image in supabase
+    response = supabase.storage.from_('generated_images').upload(image_path, file)
+
+
 
     # Construct the URL to access the image
     space_url = "https://killwithabass-flux-1-dev-lora-androflux.hf.space"  # Replace with your actual space URL

requirements.txt

@@ -7,4 +7,5 @@ sentencepiece
 pillow
 climage
 pygsheets
-gradio_client
+gradio_client
+supabase