rajistics commited on
Commit
cac1b0e
·
1 Parent(s): eae7c24

add print statements for debugging

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -29,10 +29,11 @@ def load_model(model_name):
29
  AutoTokenizer.from_pretrained(model_name),
30
  )
31
 
32
-
33
  def main():
34
 
35
  st.title("Transformers Interpet Demo App")
 
36
 
37
  image = Image.open("./images/tight@1920x_transparent.png")
38
  st.sidebar.image(image, use_column_width=True)
@@ -61,6 +62,7 @@ def main():
61
  "Choose a classification model", list(models.keys())
62
  )
63
  model, tokenizer = load_model(model_name)
 
64
  if model_name.startswith("textattack/"):
65
  model.config.id2label = {0: "NEGATIVE (0) ", 1: "POSITIVE (1)"}
66
  model.eval()
 
29
  AutoTokenizer.from_pretrained(model_name),
30
  )
31
 
32
+ print ("before main")
33
  def main():
34
 
35
  st.title("Transformers Interpet Demo App")
36
+ print ("before main")
37
 
38
  image = Image.open("./images/tight@1920x_transparent.png")
39
  st.sidebar.image(image, use_column_width=True)
 
62
  "Choose a classification model", list(models.keys())
63
  )
64
  model, tokenizer = load_model(model_name)
65
+ print ("Model loaded")
66
  if model_name.startswith("textattack/"):
67
  model.config.id2label = {0: "NEGATIVE (0) ", 1: "POSITIVE (1)"}
68
  model.eval()