asrilmurdian commited on
Commit
1e7f195
·
verified ·
1 Parent(s): 892a010

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -84,15 +84,13 @@ with st.container():
84
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
85
 
86
  headers = {"Authorization": f"Bearer {HF_TOKEN}"}
87
-
88
  # Selectbox to choose between API URLs
89
  selected_api_url = st.selectbox("Select Model", options=["bart-large-cnn", "pegasus_indonesian_base-finetune"])
90
 
91
  # Determine the selected Model
92
  if selected_api_url == "bart-large-cnn":
93
- API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
94
- else:
95
- API_URL = "https://api-inference.huggingface.co/models/thonyyy/pegasus_indonesian_base-finetune"
96
 
97
  with st.container():
98
  # Download and parse the article
 
84
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
85
 
86
  headers = {"Authorization": f"Bearer {HF_TOKEN}"}
87
+
88
  # Selectbox to choose between API URLs
89
  selected_api_url = st.selectbox("Select Model", options=["bart-large-cnn", "pegasus_indonesian_base-finetune"])
90
 
91
  # Determine the selected Model
92
  if selected_api_url == "bart-large-cnn":
93
+ API_URL = "https://api-inference.huggingface.co/models/asrilmurdian/asril-pegasus-model""
 
 
94
 
95
  with st.container():
96
  # Download and parse the article