Update app.py

app.py

@@ -1,12 +1,17 @@
 
 
 import gradio as gr
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
-from transformers import BertTokenizer, BertLMHeadModel
+#from transformers import AutoModelForSequenceClassification, AutoTokenizer
+#from transformers import BertTokenizer, BertLMHeadModel
 
 # Load pre-trained model and tokenizer
-tokenizer = BertTokenizer.from_pretrained('clinicalBERT')
-model = BertLMHeadModel.from_pretrained('clinicalBERT')
+#tokenizer = BertTokenizer.from_pretrained('clinicalBERT')
+#model = BertLMHeadModel.from_pretrained('clinicalBERT')
+
+from transformers import AutoTokenizer, AutoModel
+tokenizer = AutoTokenizer.from_pretrained("medicalai/ClinicalBERT")
+model = AutoModel.from_pretrained("medicalai/ClinicalBERT")
+
 
 # Define a function to generate text using the model
 def generate_text(input_text):