Update app.py

app.py

@@ -21,10 +21,13 @@ class Chatbot():
             'API key (not necessary for now)',
             type='password',
             help="Type in your HuggingFace API key to use this app")
+        
+        login(token=os.environ['hf_api_key'])
+        self.model_checkpoint = 'namnh113/vi-mrc-large'
 
         self.checkpoint = st.sidebar.selectbox(
             label = "Choose model",
-            options = [model_checkpoint],
+            options = [self.model_checkpoint],
             help="List available model to predict"
             )
 
@@ -107,8 +110,6 @@ class Chatbot():
                 st.error(e, icon="🚨")
 
 model = Chatbot()
-login(token=os.environ['hf_api_key'])
-model_checkpoint = 'namnh113/vi-mrc-large'
-tokenizer = transformers.AutoTokenizer.from_pretrained(model_checkpoint)
-model = MRCQuestionAnswering.from_pretrained(model_checkpoint)
+tokenizer = transformers.AutoTokenizer.from_pretrained(model.model_checkpoint)
+model = MRCQuestionAnswering.from_pretrained(model.model_checkpoint)
 model.form_data()