zijia88 commited on
Commit
6468d44
·
1 Parent(s): e205ba6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
-
4
  drainage_token = os.environ.get("drainage_token")
5
 
6
  pipeline = pipeline(task="image-classification", model="zijia88/autotrain-drainage-56552131498", use_auth_token=drainage_token)
 
1
  import gradio as gr
2
  from transformers import pipeline
3
+ import os
4
  drainage_token = os.environ.get("drainage_token")
5
 
6
  pipeline = pipeline(task="image-classification", model="zijia88/autotrain-drainage-56552131498", use_auth_token=drainage_token)