EdenSw commited on
Commit
a548147
·
1 Parent(s): a6d5cad

Update API.py

Browse files
Files changed (1) hide show
  1. API.py +17 -0
API.py CHANGED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import requests
2
+
3
+ A2T_URL = "https://api-inference.huggingface.co/models/openai/whisper-large-v2"
4
+ SUM_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
5
+ headers = {"Authorization": "Bearer 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"}
6
+
7
+
8
+ def aud2text(filename):
9
+ with open(filename, "rb") as f:
10
+ data = f.read()
11
+ response = requests.post(A2T_URL, headers=headers, data=data)
12
+ return response.json()
13
+
14
+
15
+ def Summarize(payload):
16
+ response = requests.post(SUM_URL, headers=headers, json=payload)
17
+ return response.json()