Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ For citation, please refer to the tool's documentation.
|
|
26 |
def inference_t5(input_text, selected_model):
|
27 |
model = T5ForClassification.from_pretrained("gokceuludogan/turna_tr_hateprint_w0.1_new_") #_b128")
|
28 |
tokenizer = AutoTokenizer.from_pretrained("gokceuludogan/turna_tr_hateprint_w0.1_new_") #_b128")
|
29 |
-
return model(**tokenizer(
|
30 |
|
31 |
|
32 |
# Functions for model-based tasks
|
|
|
26 |
def inference_t5(input_text, selected_model):
|
27 |
model = T5ForClassification.from_pretrained("gokceuludogan/turna_tr_hateprint_w0.1_new_") #_b128")
|
28 |
tokenizer = AutoTokenizer.from_pretrained("gokceuludogan/turna_tr_hateprint_w0.1_new_") #_b128")
|
29 |
+
return model(**tokenizer(input_text, return_tensors='pt')).logits
|
30 |
|
31 |
|
32 |
# Functions for model-based tasks
|