Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer
+from turkish_lm_tuner import T5ForClassification
 import os
 
 # Retrieve Hugging Face authentication token from environment variables
@@ -22,8 +23,17 @@ APP_CITATION = """
 For citation, please refer to the tool's documentation.
 """
 
+def inference_t5(input_text, selected_model):
+    model = T5ForClassification.from_pretrained("gokceuludogan/turna_tr_hateprint_w0.1_new_") #_b128")
+    tokenizer = AutoTokenizer.from_pretrained("gokceuludogan/turna_tr_hateprint_w0.1_new_") #_b128")
+    return model(**tokenizer(test_texts, return_tensors='pt')).logits
+
+    
 # Functions for model-based tasks
 def perform_binary_classification(input_text, selected_model):
+    if 'turna' in selected_model:
+        return inference_t5(input_text, selected_model)
+        
     model = pipeline(model=f'gokceuludogan/{selected_model}')
     return model(input_text)[0]