XGBooster commited on
Commit
365471b
·
1 Parent(s): becb2e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
 
@@ -32,7 +32,7 @@ def generate_images(prompt: str, scale: str, iterations: str, seed: str, num_ima
32
  return output_files_names
33
 
34
 
35
- def transcribe_audio(model_selected :str, audio_input: BinaryIO) -> tuple[str, str, str, str]:
36
 
37
  model = whisper.load_model(model_selected)
38
  audio_input = whisper.load_audio(audio_input)
 
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:
15
  AUTH_TOKEN = get_token()
16
  device = get_device()
17
 
 
32
  return output_files_names
33
 
34
 
35
+ def transcribe_audio(model_selected :str, audio_input: BinaryIO) -> tuple:
36
 
37
  model = whisper.load_model(model_selected)
38
  audio_input = whisper.load_audio(audio_input)