Update app.py

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
 from transformers import pipeline
 
-pipeline = pipeline(task="image-classification", model="zijia88/autotrain-drainage-56552131498",use_auth_token={drainage_token})
+drainage_token = os.environ.get("drainage_token")
 
+pipeline = pipeline(task="image-classification", model="zijia88/autotrain-drainage-56552131498", use_auth_token=drainage_token)
 def predict(image):
     predictions = pipeline(image)
     return {p["label"]: p["score"] for p in predictions}