Update app.py

app.py

@@ -94,20 +94,19 @@ def audio_to_text():
 
 def summarize():
     transcript = audio_to_text()
-    len_trans = len(transcript)
-
-    chunks = int(len_trans/512)
-    from transformers import pipeline
-
-    summarizer = pipeline("summarization", model="Azma-AI/bart-large-text-summarizer")
-
+    import requests
     
-    #print(summarizer(transcript, do_sample=False))
-    cutoff = 512
-    final_output = ''
+    API_URL = "https://api-inference.huggingface.co/models/Azma-AI/bart-large-text-summarizer"
+    headers = {"Authorization": f"Bearer {API_TOKEN}"}
     
-    final_output = summarizer(transcript, do_sample=False)
-    return final_output
+    def query(payload):
+    	response = requests.post(API_URL, headers=headers, json=payload)
+    	return response.json()
+    	
+    output = query({
+    	"inputs": transcript["text"],
+    })
+    return output
 
 yt_link = st.text_input("Enter the YouTube URL: ")