Update app.py

app.py

@@ -64,11 +64,18 @@ if fetch_button:
 # HuggingFace API KEY input
 API_KEY = st.text_input("Enter your HuggingFace API key", type="password")
 
-# HuggingFace API inference URL.
-API_URL = "https://api-inference.huggingface.co/models/google/pegasus-cnn_dailymail"
 
-headers = {"Authorization": f"Bearer {API_KEY}"}
+# Selectbox to choose between API URLs
+selected_api_url = st.selectbox("Select Model", options=["pegasus-cnn_dailymail", "pegasus_indonesian_base-finetune"])
+
+# Determine the selected Model
+if selected_api_url == "pegasus-cnn_dailymail":
+    API_URL = "https://api-inference.huggingface.co/models/google/pegasus-cnn_dailymail"
+else:
+    API_URL = "https://api-inference.huggingface.co/models/thonyyy/pegasus_indonesian_base-finetune"
+
 
+headers = {"Authorization": f"Bearer {API_KEY}"}
 
 submit_button = st.button("Submit")
 
@@ -80,6 +87,7 @@ if submit_button:
 
     title = article.title
     text = article.text
+    summary_lib = article.summary
 
     # HuggingFace API request function
     def query(payload):
@@ -96,6 +104,14 @@ if submit_button:
         summary = output[0]['summary_text'].replace('<n>', " ") 
 
         st.divider()
-        st.subheader("Summary")
+        st.subheader("Summary AI")
         st.write(f"Your article: **{title}**")
-        st.write(f"**{summary}**")
+        st.write(f"**{summary}**")
+        
+        st.divider()
+        st.subheader("Summary Library Newspaper")
+        st.write(f"**{summary_lib}**")
+        
+        st.divider()
+        st.subheader("Real Article")
+        st.write(f"**{text}**")