Commit
•
3522874
1
Parent(s):
551980e
Update src/v2_for_hf.py
Browse files- src/v2_for_hf.py +1 -1
src/v2_for_hf.py
CHANGED
|
@@ -12,7 +12,7 @@ from huggingface_hub import login
|
|
| 12 |
login(token=os.environ.get("HF_token"))
|
| 13 |
|
| 14 |
# Modfiy this to change the number of generations
|
| 15 |
-
NUM_GEN =
|
| 16 |
|
| 17 |
def encode_image(image_path):
|
| 18 |
with open(image_path, "rb") as image_file:
|
|
|
|
| 12 |
login(token=os.environ.get("HF_token"))
|
| 13 |
|
| 14 |
# Modfiy this to change the number of generations
|
| 15 |
+
NUM_GEN = 3
|
| 16 |
|
| 17 |
def encode_image(image_path):
|
| 18 |
with open(image_path, "rb") as image_file:
|