gokceuludogan commited on
Commit
31487be
·
verified ·
1 Parent(s): 81fc300

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(test_texts, return_tensors='pt')).logits
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