Spaces:
Running
Running
Commit
·
940a46c
1
Parent(s):
379e091
added error handling
Browse files
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")
|