Spaces:
Sleeping
Sleeping
Andre
commited on
Commit
·
ef908f6
1
Parent(s):
472af83
“Update”
Browse files- app.py +1 -0
- src/img_gen.py +1 -7
app.py
CHANGED
@@ -10,6 +10,7 @@ if src_path not in sys.path:
|
|
10 |
|
11 |
# Import gradio_interface
|
12 |
from gradio_interface import demo
|
|
|
13 |
|
14 |
|
15 |
if __name__ == "__main__":
|
|
|
10 |
|
11 |
# Import gradio_interface
|
12 |
from gradio_interface import demo
|
13 |
+
from config.config import api_token
|
14 |
|
15 |
|
16 |
if __name__ == "__main__":
|
src/img_gen.py
CHANGED
@@ -1,18 +1,12 @@
|
|
1 |
# img_gen.py
|
2 |
import sys
|
3 |
import os
|
4 |
-
|
5 |
-
# Add the src folder to the Python path
|
6 |
-
# Solves all problems w subfolders - option2
|
7 |
-
src_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "src"))
|
8 |
-
if src_path not in sys.path:
|
9 |
-
sys.path.append(src_path)
|
10 |
-
|
11 |
import random
|
12 |
from huggingface_hub import InferenceClient
|
13 |
from datetime import datetime
|
14 |
from config.config import api_token, models, prompts # Direct import
|
15 |
|
|
|
16 |
def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
|
17 |
try:
|
18 |
# Generate the image
|
|
|
1 |
# img_gen.py
|
2 |
import sys
|
3 |
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
import random
|
5 |
from huggingface_hub import InferenceClient
|
6 |
from datetime import datetime
|
7 |
from config.config import api_token, models, prompts # Direct import
|
8 |
|
9 |
+
|
10 |
def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
|
11 |
try:
|
12 |
# Generate the image
|