Update app.py

app.py

@@ -4,40 +4,41 @@ import requests, json
 from transformers import pipeline
 import gradio as gr
 
+load_dotenv()
 # Access the Hugging Face API key and endpoint URL
 hf_api_key = os.getenv('HF_API_KEY')
-ENDPOINT_URL = os.getenv('HF_API_SUMMARY_BASE')
+MODEL_NAME = "sshleifer/distilbart-cnn-12-6"
+ENDPOINT_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 
 #Summarization endpoint
-def get_completion(inputs, parameters=None, ENDPOINT_URL=ENDPOINT_URL): 
+def get_completion(inputs, parameters=None):
     headers = {
-      "Authorization": f"Bearer {hf_api_key}",
-      "Content-Type": "application/json"
+        "Authorization": f"Bearer {hf_api_key}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "inputs": inputs
     }
-    data = { "inputs": inputs }
     if parameters is not None:
         data.update({"parameters": parameters})
     
     try:
-        response = requests.request("POST",
-                                ENDPOINT_URL, headers=headers,
-                                data=json.dumps(data)
-                               )
-        if response.status_code == 200:
-            return json.loads(response.content.decode("utf-8"))
-        else:
-            print(f"Error: Received status code {response.status_code}")
-            return {"error": f"Bad response: {response.status_code}"}
+        response = requests.post(ENDPOINT_URL, headers=headers, json=data)
+        response.raise_for_status()
+        return response.json()
     except requests.exceptions.RequestException as e:
         print(f"Request failed: {e}")
-        return {"error": "Request failed"}
-    except json.JSONDecodeError as e:
-        print(f"JSON decoding failed: {e}")
-        return {"error": "Invalid JSON response"}
+        return {"error": f"Request failed: {str(e)}"}
 
-def summarize(input):
-    output = get_completion(input)
-    return output[0]['summary_text']
+def summarize(input_text):
+    try:
+        output = get_completion(input_text)
+        if isinstance(output, list) and len(output) > 0 and 'summary_text' in output[0]:
+            return output[0]['summary_text']
+        else:
+            return f"Unexpected response format: {output}"
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
 
 demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
 demo.launch()