Spaces:
Sleeping
Sleeping
updated
Browse files
app.py
CHANGED
@@ -5,14 +5,9 @@ import numpy as np
|
|
5 |
|
6 |
|
7 |
# Define the function for sentiment analysis
|
8 |
-
@st.cache_resource()
|
9 |
|
10 |
-
|
11 |
-
|
12 |
-
tokenizer = AutoTokenizer.from_pretrained("xlnet-base-cased")
|
13 |
-
model = AutoModelForSequenceClassification.from_pretrained("MrDdz/mytuned_test_trainer-base-cased1")
|
14 |
-
return tokenizer,model
|
15 |
-
tokenizer, model = get_model()
|
16 |
|
17 |
# # Setting the page configurations
|
18 |
# st.set_page_config(
|
@@ -66,6 +61,15 @@ unsafe_allow_html=True
|
|
66 |
# else:
|
67 |
# st.warning(f"The sentiment is {sentiment} with a score of {score*100:.2f}%!")
|
68 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
69 |
if text:
|
70 |
text_sample = tokenizer([text], padding = True, Truncation = True, max_length = 512, return_tensors = 'pt' )
|
71 |
output = model(** text_sample)
|
|
|
5 |
|
6 |
|
7 |
# Define the function for sentiment analysis
|
|
|
8 |
|
9 |
+
|
10 |
+
|
|
|
|
|
|
|
|
|
11 |
|
12 |
# # Setting the page configurations
|
13 |
# st.set_page_config(
|
|
|
61 |
# else:
|
62 |
# st.warning(f"The sentiment is {sentiment} with a score of {score*100:.2f}%!")
|
63 |
|
64 |
+
|
65 |
+
@st.cache_resource()
|
66 |
+
def get_model():
|
67 |
+
# Load the model and tokenizer
|
68 |
+
tokenizer = AutoTokenizer.from_pretrained("xlnet-base-cased")
|
69 |
+
model = AutoModelForSequenceClassification.from_pretrained("MrDdz/mytuned_test_trainer-base-cased1")
|
70 |
+
return tokenizer,model
|
71 |
+
tokenizer, model = get_model()
|
72 |
+
|
73 |
if text:
|
74 |
text_sample = tokenizer([text], padding = True, Truncation = True, max_length = 512, return_tensors = 'pt' )
|
75 |
output = model(** text_sample)
|