Andre commited on
Commit
ef908f6
·
1 Parent(s): 472af83

“Update”

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. 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