Spaces:
Runtime error
Runtime error
Commit
·
585c098
1
Parent(s):
3b89cb9
Update app.py
Browse files
app.py
CHANGED
@@ -149,16 +149,15 @@ import tokenizers # Import tokenizers explicitly
|
|
149 |
model = transformers.AutoModelForSequenceClassification.from_pretrained("DeeeTeeee01/mytest_trainer_roberta-base")
|
150 |
tokenizer = transformers.AutoTokenizer.from_pretrained("DeeeTeeee01/mytest_trainer_roberta-base")
|
151 |
|
152 |
-
# Define the custom hash function for tokenizers.Tokenizer
|
153 |
def tokenizer_hash_func(tokenizer):
|
154 |
return tokenizer.__str__()
|
155 |
|
156 |
-
# Define the custom hash function for tokenizers.AddedToken
|
157 |
def added_token_hash_func(token):
|
158 |
return token.__str__()
|
159 |
|
160 |
-
# Define the function for sentiment analysis
|
161 |
-
@st.cache_data(hash_funcs={tokenizers.Tokenizer: tokenizer_hash_func, tokenizers.AddedToken: added_token_hash_func})
|
162 |
def predict_sentiment(text):
|
163 |
# Load the pipeline
|
164 |
pipeline = transformers.pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
@@ -215,4 +214,4 @@ h1 {
|
|
215 |
</style>
|
216 |
""",
|
217 |
unsafe_allow_html=True
|
218 |
-
)
|
|
|
149 |
model = transformers.AutoModelForSequenceClassification.from_pretrained("DeeeTeeee01/mytest_trainer_roberta-base")
|
150 |
tokenizer = transformers.AutoTokenizer.from_pretrained("DeeeTeeee01/mytest_trainer_roberta-base")
|
151 |
|
152 |
+
# Define the custom hash function for tokenizers.Tokenizer (Unused in this version)
|
153 |
def tokenizer_hash_func(tokenizer):
|
154 |
return tokenizer.__str__()
|
155 |
|
156 |
+
# Define the custom hash function for tokenizers.AddedToken (Unused in this version)
|
157 |
def added_token_hash_func(token):
|
158 |
return token.__str__()
|
159 |
|
160 |
+
# Define the function for sentiment analysis without caching
|
|
|
161 |
def predict_sentiment(text):
|
162 |
# Load the pipeline
|
163 |
pipeline = transformers.pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
|
|
214 |
</style>
|
215 |
""",
|
216 |
unsafe_allow_html=True
|
217 |
+
)
|