Update app.py

app.py

@@ -1,23 +1,30 @@
-# Helper function
+### Helper function
 import os, requests, json
 
 hf_api_key = os.environ['HF_API_KEY']
+ENDPOINT_URL=os.environ['HF_API_SUMMARY_BASE']
 
 #Summarization endpoint
-def get_completion(inputs, parameters=None, ENDPOINT_URL="shleifer/distilbart-cnn-12-6"): 
+def get_completion(inputs, parameters=None):
+    
     headers = {
       "Authorization": f"Bearer {hf_api_key}",
       "Content-Type": "application/json"
     }
-    data = { "inputs": inputs }
+    
+    payload = { "inputs": inputs }
+    
     if parameters is not None:
-        data.update({"parameters": parameters})
-    response = requests.request("POST",
-                                ENDPOINT_URL, headers=headers,
-                                data=json.dumps(data)
-                               )
+        payload.update({"parameters": parameters})
+    
+    response = requests.post(ENDPOINT_URL, 
+                             headers=headers,
+                             data=json.dumps(payload))
+    
     return json.loads(response.content.decode("utf-8"))
 
+### Gradio App
+
 import gradio as gr
 
 def summarize(input):