ashhadahsan commited on
Commit
940a46c
·
1 Parent(s): 379e091

added error handling

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -34,7 +34,7 @@ model_classes ={
34
  13: "WiFi",
35
  }
36
 
37
- @st.cache(suppress_st_warning=True)
38
  def load_t5():
39
  model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
40
 
@@ -42,23 +42,23 @@ def load_t5():
42
  return model, tokenizer
43
 
44
 
45
- @st.cache(suppress_st_warning=True)
46
  def custom_model():
47
  return pipeline("summarization", model="my_awesome_sum/")
48
 
49
 
50
- @st.cache(suppress_st_warning=True)
51
  def convert_df(df):
52
  # IMPORTANT: Cache the conversion to prevent computation on every rerun
53
  return df.to_csv(index=False).encode("utf-8")
54
 
55
 
56
- @st.cache(suppress_st_warning=True)
57
  def load_one_line_summarizer(model):
58
  return model.load_model("t5", "snrspeaks/t5-one-line-summary")
59
 
60
 
61
- @st.cache(suppress_st_warning=True)
62
  def classify_category():
63
  tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
64
  new_model = load_model("model")
 
34
  13: "WiFi",
35
  }
36
 
37
+ @st.cache(suppress_st_warning=True,suppress_st_warning=False)
38
  def load_t5():
39
  model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
40
 
 
42
  return model, tokenizer
43
 
44
 
45
+ @st.cache(suppress_st_warning=True,suppress_st_warning=False)
46
  def custom_model():
47
  return pipeline("summarization", model="my_awesome_sum/")
48
 
49
 
50
+ @st.cache(suppress_st_warning=True,suppress_st_warning=False)
51
  def convert_df(df):
52
  # IMPORTANT: Cache the conversion to prevent computation on every rerun
53
  return df.to_csv(index=False).encode("utf-8")
54
 
55
 
56
+ @st.cache(suppress_st_warning=True,suppress_st_warning=False)
57
  def load_one_line_summarizer(model):
58
  return model.load_model("t5", "snrspeaks/t5-one-line-summary")
59
 
60
 
61
+ @st.cache(suppress_st_warning=True,suppress_st_warning=False)
62
  def classify_category():
63
  tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
64
  new_model = load_model("model")