Update app.py

app.py

@@ -18,7 +18,7 @@ from transformers import pipeline
 # Load model
 #model = TFBertForSequenceClassification.from_pretrained('https://huggingface.co/spaces/Kleo/Sarcasm/tree/main')
 
-pipeline= pipeline(task="sequence-classification", model="https://huggingface.co/spaces/Kleo/Sarcasm/blob/main/fine-tuned-bert-gr.h5")
+pipeline= pipeline(task="text-classification", model="https://huggingface.co/spaces/Kleo/Sarcasm/blob/main/fine-tuned-bert-gr.h5")
 def check_sarcasm(sentence):
     #tf_batch = tokenizer(sentence, max_length=128, padding=True, truncation=True, return_tensors='tf')
     #tf_outputs = model(tf_batch.input_ids, tf_batch.token_type_ids)