Update app.py

app.py

@@ -6,22 +6,15 @@ from transformers import AutoModelForTokenClassification
 from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
 
 
-# Define hyperparameters for QnA
-learning_rate = 3e-5
-batch_size = 16
-epochs = 3
+# QnA Model (Context based)
 max_seq_length = 512
-warmup_steps = 100
-weight_decay = 0.01
-dropout_prob = 0.1
-gradient_clip_value = 1.0
 q_n_a_model_name = "deepset/roberta-base-squad2"
 q_n_a_model = AutoModelForQuestionAnswering.from_pretrained(q_n_a_model_name)
 q_n_a_tokenizer = AutoTokenizer.from_pretrained(q_n_a_model_name)  
 
-
-classification_tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
-classification_model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased")
+classification_model_name = "distilbert-base-uncased"
+classification_tokenizer = DistilBertTokenizer.from_pretrained(classification_model_name)
+classification_model = DistilBertForSequenceClassification.from_pretrained(classification_model_name)