Spaces:
Runtime error
Runtime error
macrdel
commited on
Commit
•
7338b09
1
Parent(s):
75483f2
update pipeline summarize src.py
Browse files- app/src/src.py +3 -3
app/src/src.py
CHANGED
@@ -62,7 +62,7 @@ def pipeline_stats(data):
|
|
62 |
"""Get statistic of sentiment"""
|
63 |
return data['sentiment'].value_counts(normalize=True).mul(100).round(2)
|
64 |
|
65 |
-
def pipeline_summarize(data, headers, url, length=2000, max_length=
|
66 |
"""Get summarization result"""
|
67 |
text = " ".join(data)
|
68 |
result_text = []
|
@@ -76,6 +76,6 @@ def pipeline_summarize(data, headers, url, length=2000, max_length=50):
|
|
76 |
}
|
77 |
}
|
78 |
res = requests.post(url, headers=headers, json=payload)
|
79 |
-
result_text.append(res.json())
|
80 |
|
81 |
-
return ". ".join(
|
|
|
62 |
"""Get statistic of sentiment"""
|
63 |
return data['sentiment'].value_counts(normalize=True).mul(100).round(2)
|
64 |
|
65 |
+
def pipeline_summarize(data, headers, url, length=2000, max_length=35):
|
66 |
"""Get summarization result"""
|
67 |
text = " ".join(data)
|
68 |
result_text = []
|
|
|
76 |
}
|
77 |
}
|
78 |
res = requests.post(url, headers=headers, json=payload)
|
79 |
+
result_text.append(res.json()[0]["generated_text"])
|
80 |
|
81 |
+
return ". ".join(result_text)
|