Spaces:
Bradarr
/
Running on Zero

Bradarr commited on
Commit
107a152
·
verified ·
1 Parent(s): 93d52a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ from watermarking import watermark
10
  import whisper
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
  import logging
 
13
 
14
  # Configure logging
15
  logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
10
  import whisper
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
  import logging
13
+ from transformers import GenerationConfig
14
 
15
  # Configure logging
16
  logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')