Update app.py

app.py

@@ -2,6 +2,10 @@ import streamlit as st
 import requests
 import json
 import time
+import os
+api_url = os.getenv("API_URL")
+auth_token = os.getenv("AUTH_TOKEN")
+
 
 # Set page config
 st.set_page_config(
@@ -90,6 +94,11 @@ if st.button("Summarize"):
     if thread_url.strip():
         with st.spinner("Analyzing thread..."):
             # Build request payload
+            headers = {
+                'Content-Type': 'application/json',
+                'Authorization': f'Bearer {auth_token}'
+            }
+
             payload = {
                 "inputs": [
                     {
@@ -98,7 +107,6 @@ if st.button("Summarize"):
                         "data": [thread_url],
                         "datatype": "BYTES"
                     },
-                    # Optional parameters
                     {
                         "name": "temperature",
                         "optional": True,
@@ -110,14 +118,14 @@ if st.button("Summarize"):
                         "name": "top_p",
                         "optional": True,
                         "shape": [1],
-                        "data": [1.0],
+                        "data": [1],
                         "datatype": "FP64"
                     },
                     {
                         "name": "repetition_penalty",
                         "optional": True,
                         "shape": [1],
-                        "data": [1.0],
+                        "data": [1],
                         "datatype": "FP64"
                     },
                     {
@@ -144,13 +152,12 @@ if st.button("Summarize"):
                 ]
             }
 
+            
+            
+            
             try:
                 # Send POST request to your local model server
-                response = requests.post(
-                    "http://localhost:8000/v2/models/inferless-model/infer",
-                    json=payload,
-                    timeout=30
-                )
+                response = requests.post(api_url, headers=headers, json=payload,timeout=30)
                 response.raise_for_status()  # Raise HTTPError if status != 200
 
                 # Parse JSON response
@@ -164,4 +171,4 @@ if st.button("Summarize"):
             except requests.exceptions.RequestException as e:
                 st.error(f"Error calling the model API: {e}")
     else:
-        st.error("Please enter a valid URL.")
+        st.error("Please enter a valid URL.")