Andre commited on
Commit
c39fbad
·
1 Parent(s): ece139c
Files changed (2) hide show
  1. config.py +1 -2
  2. img_gen_logic.py +0 -1
config.py CHANGED
@@ -1,7 +1,6 @@
1
  # config.py
2
  import os
3
- #from google.colab import userdata # Only for Colab
4
- ##
5
 
6
  # Retrieve the Hugging Face token
7
  #try:
 
1
  # config.py
2
  import os
3
+ from prompts import prompts # Import prompts from prompts.py
 
4
 
5
  # Retrieve the Hugging Face token
6
  #try:
img_gen_logic.py CHANGED
@@ -4,7 +4,6 @@ from huggingface_hub import InferenceClient
4
  from PIL import Image
5
  from datetime import datetime
6
  from config import api_token, models, prompts # Direct import
7
- from prompts import prompts # Import prompts from prompts.py
8
 
9
 
10
  def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
 
4
  from PIL import Image
5
  from datetime import datetime
6
  from config import api_token, models, prompts # Direct import
 
7
 
8
 
9
  def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):