XGBooster commited on
Commit
becb2e9
·
1 Parent(s): 3463cdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def get_device() -> Literal['cuda', 'cpu']:
11
  def get_token() -> str:
12
  return os.environ.get("HUGGING_FACE_TOKEN")
13
 
14
- def generate_images(prompt: str, scale: str, iterations: str, seed: str, num_images: str) -> list[str]:
15
  AUTH_TOKEN = get_token()
16
  device = get_device()
17
 
 
11
  def get_token() -> str:
12
  return os.environ.get("HUGGING_FACE_TOKEN")
13
 
14
+ def generate_images(prompt: str, scale: str, iterations: str, seed: str, num_images: str) -> list(str):
15
  AUTH_TOKEN = get_token()
16
  device = get_device()
17