Update app.py

app.py

@@ -1,10 +1,14 @@
 import gradio as gr
-from transformers import pipeline
 import re
-from keybert import KeyBERT
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
+from keybert import KeyBERT
 
-text_pipe = pipeline("text-classification", model="karalif/myTestModel", return_all_scores=True)
+# Initialize your model and tokenizer here
+# Replace 'your_model_path' and 'your_tokenizer_path' with the actual paths or model identifiers
+model_identifier = "karalif/myTestModel"
+new_model = AutoModelForSequenceClassification.from_pretrained(model_identifier)
+new_tokenizer = AutoTokenizer.from_pretrained(model_identifier)
 
 def get_prediction(text):
     # Tokenize the input text
@@ -68,4 +72,4 @@ demo = gr.Interface(
     theme=gr.themes.Default(primary_hue="red", secondary_hue="pink")
 )
 
-demo.launch()
+demo.launch()