Spaces:
Running
Running
Update app/main.py
Browse files- app/main.py +4 -9
app/main.py
CHANGED
@@ -6,7 +6,6 @@ import re
|
|
6 |
import os
|
7 |
|
8 |
token = os.environ.get("huggingface_token")
|
9 |
-
key = os.environ.get("rapid_key")
|
10 |
|
11 |
app = FastAPI()
|
12 |
|
@@ -54,7 +53,7 @@ async def root(item: Item):
|
|
54 |
}
|
55 |
headers = {
|
56 |
"content-type": "application/json",
|
57 |
-
"X-RapidAPI-Key":
|
58 |
"X-RapidAPI-Host": "text-analysis12.p.rapidapi.com"
|
59 |
}
|
60 |
|
@@ -106,23 +105,19 @@ async def root(item: Item):
|
|
106 |
|
107 |
article = extract_article_content(item.url)
|
108 |
|
109 |
-
|
110 |
|
111 |
API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
|
112 |
headers = {'Authorization': token}
|
113 |
|
114 |
-
|
115 |
-
def query(payload):
|
116 |
-
response = requests.post(API_URL, headers=headers, json=payload)
|
117 |
-
return response.json()
|
118 |
|
119 |
params = {'do_sample': False}
|
120 |
|
121 |
-
|
122 |
-
output = query({
|
123 |
'inputs': article,
|
124 |
'parameters': params
|
125 |
})
|
|
|
|
|
126 |
|
127 |
return {"summary": output}
|
128 |
|
|
|
6 |
import os
|
7 |
|
8 |
token = os.environ.get("huggingface_token")
|
|
|
9 |
|
10 |
app = FastAPI()
|
11 |
|
|
|
53 |
}
|
54 |
headers = {
|
55 |
"content-type": "application/json",
|
56 |
+
"X-RapidAPI-Key":"",
|
57 |
"X-RapidAPI-Host": "text-analysis12.p.rapidapi.com"
|
58 |
}
|
59 |
|
|
|
105 |
|
106 |
article = extract_article_content(item.url)
|
107 |
|
|
|
108 |
|
109 |
API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
|
110 |
headers = {'Authorization': token}
|
111 |
|
|
|
|
|
|
|
|
|
112 |
|
113 |
params = {'do_sample': False}
|
114 |
|
115 |
+
response = requests.post(API_URL, headers=headers, json={
|
|
|
116 |
'inputs': article,
|
117 |
'parameters': params
|
118 |
})
|
119 |
+
|
120 |
+
output = response.json()
|
121 |
|
122 |
return {"summary": output}
|
123 |
|