Update app.py

app.py

@@ -14,29 +14,25 @@ load_dotenv()
 
 # Access the Hugging Face API key and endpoint URL
 hf_api_key = os.getenv('HF_API_KEY')
-MODEL_NAME = "sshleifer/distilbart-cnn-12-6"
+MODEL_NAME = "facebook/bart-large-cnn"
 ENDPOINT_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 
-def get_completion(inputs, parameters=None):
+def get_completion(inputs):
     headers = {
         "Authorization": f"Bearer {hf_api_key}",
         "Content-Type": "application/json"
     }
-    data = {
+    payload = {
         "inputs": inputs,
-        "options": {"wait_for_model": True}
+        "parameters": {"max_length": 130, "min_length": 30}
     }
-    if parameters is not None:
-        data.update({"parameters": parameters})
     
     logger.debug(f"Sending request to {ENDPOINT_URL}")
-    logger.debug(f"Request headers: {headers}")
-    logger.debug(f"Request data: {json.dumps(data, indent=2)}")
+    logger.debug(f"Request payload: {json.dumps(payload, indent=2)}")
 
     try:
-        response = requests.post(ENDPOINT_URL, headers=headers, json=data)
+        response = requests.post(ENDPOINT_URL, headers=headers, json=payload)
         logger.debug(f"Response status code: {response.status_code}")
-        logger.debug(f"Response headers: {dict(response.headers)}")
         logger.debug(f"Response content: {response.text}")
         
         response.raise_for_status()
@@ -51,7 +47,7 @@ def summarize(input_text):
     try:
         logger.info(f"Received input text: {input_text[:100]}...")  # Log first 100 chars of input
         output = get_completion(input_text)
-        if isinstance(output, list) and len(output) > 0 and 'summary_text' in output[0]:
+        if isinstance(output, list) and len(output) > 0 and isinstance(output[0], dict) and 'summary_text' in output[0]:
             return output[0]['summary_text']
         elif isinstance(output, dict) and 'error' in output:
             return f"API Error: {output['error']}"