merve HF Staff commited on
Commit
9508bc0
·
1 Parent(s): 3f20002

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from streamlit_chat import message
5
 
6
 
7
  api_token = os.getenv("api_token")
8
-
9
  @st.cache
10
  def query(payload, model_id, api_token):
11
  headers = {"Authorization": f"Bearer {api_token}"}
 
5
 
6
 
7
  api_token = os.getenv("api_token")
8
+ model_id = "deepset/roberta-base-squad2"
9
  @st.cache
10
  def query(payload, model_id, api_token):
11
  headers = {"Authorization": f"Bearer {api_token}"}