baconnier commited on
Commit
9d5ac59
1 Parent(s): d42af76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -281,6 +281,7 @@ class GradioInterface:
281
 
282
 
283
  if __name__ == '__main__':
 
284
  prompt_refiner = PromptRefiner(api_token)
285
  gradio_interface = GradioInterface(prompt_refiner)
286
  gradio_interface.launch(share=True)
 
281
 
282
 
283
  if __name__ == '__main__':
284
+ api_token = os.getenv('HF_API_TOKEN')
285
  prompt_refiner = PromptRefiner(api_token)
286
  gradio_interface = GradioInterface(prompt_refiner)
287
  gradio_interface.launch(share=True)