Update app.py

app.py

@@ -16,9 +16,9 @@ st.sidebar.image(image, caption=None, width=None, use_column_width=None, clamp=F
 def load_model(version):
     if version == "QAmemBERT1-base":
         return pipeline('question-answering', model='CATIE-AQ/QAmembert', tokenizer='CATIE-AQ/QAmembert')
-    elif version == "QAmemBERT2":
+    if version == "QAmemBERT2":
         return pipeline('question-answering', model='CATIE-AQ/QAmembert2', tokenizer='CATIE-AQ/QAmembert2')
-    elif version == "QAmemBERTa":
+    if version == "QAmemBERTa":
         return pipeline('question-answering', model='CATIE-AQ/QAmemberta', tokenizer='CATIE-AQ/QAmemberta')
     else:
         return pipeline('question-answering', model='CATIE-AQ/QAmembert-large', tokenizer='CATIE-AQ/QAmembert-large')