Spaces:
Sleeping
Sleeping
Update generation.py
Browse files- generation.py +1 -1
generation.py
CHANGED
|
@@ -8,7 +8,7 @@ OUTPUT_DIR = "generated_images"
|
|
| 8 |
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
| 9 |
|
| 10 |
# Usaremos un modelo de la comunidad que suele ser gratuito
|
| 11 |
-
client = InferenceClient()
|
| 12 |
|
| 13 |
def generate_image_from_prompt(
|
| 14 |
prompt: str,
|
|
|
|
| 8 |
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
| 9 |
|
| 10 |
# Usaremos un modelo de la comunidad que suele ser gratuito
|
| 11 |
+
client = InferenceClient(token=os.getenv("HF_TOKEN"))
|
| 12 |
|
| 13 |
def generate_image_from_prompt(
|
| 14 |
prompt: str,
|