Andre commited on
Commit
959fd70
·
1 Parent(s): e22b115

“Update”

Browse files
Files changed (1) hide show
  1. src/img_gen.py +1 -1
src/img_gen.py CHANGED
@@ -4,7 +4,7 @@ import os
4
  import random
5
  from huggingface_hub import InferenceClient
6
  from datetime import datetime
7
- from config.config import models, prompts # Direct import
8
 
9
  def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
10
  try:
 
4
  import random
5
  from huggingface_hub import InferenceClient
6
  from datetime import datetime
7
+ from config.config import models, prompts, api_token # Direct import
8
 
9
  def generate(prompt_alias, team, model_alias, custom_prompt, height=360, width=640, num_inference_steps=20, guidance_scale=2.0, seed=-1):
10
  try: