Add Flagging

app.py

@@ -1,9 +1,11 @@
 from typing import Any, Sequence
+
 import gradio as gr
+import tensorflow as tf
+import torch
 from gradio import CSVLogger, FlaggingCallback
 from gradio.components import Component
-import torch
-import tensorflow as tf
+from transformers import DebertaV2Tokenizer, TFAutoModelForSequenceClassification, AutoModelForSequenceClassification
 
 USE_TENSORFLOW = True
 
@@ -14,7 +16,9 @@ CLASSES = {
     'no': 2,
 }
 tokenizer = DebertaV2Tokenizer.from_pretrained('cross-encoder/nli-deberta-v3-base', do_lower_case=True)
-model = TFAutoModelForSequenceClassification.from_pretrained('MrPio/TheSeagullStory-nli-deberta-v3-base',dtype=tf.float16) if USE_TENSORFLOW else AutoModelForSequenceClassification.from_pretrained('MrPio/TheSeagullStory-nli-deberta-v3-base')
+model = TFAutoModelForSequenceClassification.from_pretrained('MrPio/TheSeagullStory-nli-deberta-v3-base',
+                                                             dtype=tf.float16) if USE_TENSORFLOW else AutoModelForSequenceClassification.from_pretrained(
+    'MrPio/TheSeagullStory-nli-deberta-v3-base')
 if not USE_TENSORFLOW:
     model.eval()
     if torch.cuda.is_available():