Vinit710 commited on
Commit
c684162
·
verified ·
1 Parent(s): 18630f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,8 +4,9 @@ import requests
4
  import os
5
 
6
  # HF model to use (small + free)
7
- MODEL_ID = "google/flan-t5-base"
8
  API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
 
9
  API_TOKEN = st.secrets["HUGGINGFACEHUB_API_TOKEN"]
10
 
11
  # Setup headers
 
4
  import os
5
 
6
  # HF model to use (small + free)
7
+ MODEL_ID = "mistralai/Mistral-7B-Instruct-v0.1"
8
  API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
9
+
10
  API_TOKEN = st.secrets["HUGGINGFACEHUB_API_TOKEN"]
11
 
12
  # Setup headers