Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -57,13 +57,7 @@ classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de
|
|
57 |
|
58 |
|
59 |
def thesis_prediction(input):
|
60 |
-
|
61 |
-
|
62 |
-
# Deserialization of the file
|
63 |
-
#file = open(path + os.path.sep + 'classIndexAssociation.pkl', 'rb')
|
64 |
-
#new_model = pickle.load(file)
|
65 |
-
|
66 |
-
#sizeOfClass = len(new_model)
|
67 |
|
68 |
model = AutoModelForSequenceClassification.from_pretrained(
|
69 |
'hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False,
|
|
|
57 |
|
58 |
|
59 |
def thesis_prediction(input):
|
60 |
+
tokenizer = AutoTokenizer.from_pretrained('"hiiamsid/BETO_es_binary_classification"', use_fast=False)
|
|
|
|
|
|
|
|
|
|
|
|
|
61 |
|
62 |
model = AutoModelForSequenceClassification.from_pretrained(
|
63 |
'hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False,
|