Update app.py

app.py

@@ -3,15 +3,11 @@ from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
 import gradio as gr
 import numpy as np
 from scipy.special import softmax
-from sklearn import metrics
-import matplotlib.pyplot as plt
-from scipy.interpolate import interp1d
-import pandas as pd
 
 tokenizer = AutoTokenizer.from_pretrained('roberta-base')
 
-from transformers import AutoModel
-model = AutoModel.from_pretrained("ilan541/OncUponTim")
+from transformers import TFAutoModelForSequenceClassification
+model = TFAutoModelForSequenceClassification.from_pretrained("ilan541/OncUponTim")
 
 def predict(your_text):
   df = split_text(text)