Update app.py

app.py

@@ -1,13 +1,21 @@
 import time
 import streamlit as st
 import requests
-BERT_LARGE = "https://api-inference.huggingface.co/models/deepset/bert-large-uncased-whole-word-masking-squad2"
 API_URL = "https://api-inference.huggingface.co/models/distilbert-base-cased-distilled-squad"
 headers = {"Authorization": "Bearer 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"}
 
 
 st.title("question-answering-demo")
-
+st.selectbox('Select a model', ('distilbert-base-cased-distilled-squad', 'roberta-base-squad2-distilled', 'xlm-roberta-large-squad2', 'bert-large-cased-whole-word-masking-finetuned-squad'))
+if option == "distilbert-base-cased-distilled-squad":
+    API_URL = 'https://api-inference.huggingface.co/models/distilbert-base-cased-distilled-squad'
+elif option == 'roberta-base-squad2-distilled':
+    API_URL = 'https://api-inference.huggingface.co/models/deepset/roberta-base-squad2-distilled'
+elif option == 'xlm-roberta-large-squad2':
+    API_URL = "https://api-inference.huggingface.co/models/deepset/xlm-roberta-large-squad2"
+if option == 'bert-large-cased-whole-word-masking-finetuned-squad':
+    API_URL = 'https://api-inference.huggingface.co/models/bert-large-cased-whole-word-masking-finetuned-squad'
+st.write(API_URL)
 text_input = st.text_area("Enter some context👇")
 text_question = st.text_input("Enter a question regarding that context👇")