Spaces:
Runtime error
Runtime error
shaocongma
commited on
Commit
·
78f4b41
1
Parent(s):
194e5ab
fix bug.
Browse files- utils/references.py +0 -2
- worker.py +2 -1
utils/references.py
CHANGED
|
@@ -209,8 +209,6 @@ def search_paper_ss(title):
|
|
| 209 |
headers = {"Accept": "*/*"}
|
| 210 |
response = requests.get(url, headers=headers, timeout=30)
|
| 211 |
results = response.json()
|
| 212 |
-
if results['total'] == 0:
|
| 213 |
-
return {}
|
| 214 |
raw_paper = results['data'][0]
|
| 215 |
if raw_paper['tldr'] is not None:
|
| 216 |
abstract = raw_paper['tldr']['text']
|
|
|
|
| 209 |
headers = {"Accept": "*/*"}
|
| 210 |
response = requests.get(url, headers=headers, timeout=30)
|
| 211 |
results = response.json()
|
|
|
|
|
|
|
| 212 |
raw_paper = results['data'][0]
|
| 213 |
if raw_paper['tldr'] is not None:
|
| 214 |
abstract = raw_paper['tldr']['text']
|
worker.py
CHANGED
|
@@ -110,6 +110,7 @@ def modify_status(task_id, new_status):
|
|
| 110 |
# Close the session
|
| 111 |
session.close()
|
| 112 |
|
|
|
|
| 113 |
#######################################################################################################################
|
| 114 |
# Pipline
|
| 115 |
#######################################################################################################################
|
|
@@ -154,7 +155,7 @@ def pipeline(message_count=0, query_interval=10):
|
|
| 154 |
print("Local file path (ZIP): ", zip_path)
|
| 155 |
print("Upload to S3: ", upload_to)
|
| 156 |
upload_file(zip_path, upload_to)
|
| 157 |
-
modify_status(task_id, 2)
|
| 158 |
print(f"Success in generating the paper.")
|
| 159 |
|
| 160 |
# Complete.
|
|
|
|
| 110 |
# Close the session
|
| 111 |
session.close()
|
| 112 |
|
| 113 |
+
|
| 114 |
#######################################################################################################################
|
| 115 |
# Pipline
|
| 116 |
#######################################################################################################################
|
|
|
|
| 155 |
print("Local file path (ZIP): ", zip_path)
|
| 156 |
print("Upload to S3: ", upload_to)
|
| 157 |
upload_file(zip_path, upload_to)
|
| 158 |
+
modify_status(task_id, 2) # status: 0 - pending (default), 1 - running, 2 - completed, 3 - failed, 4 - deleted
|
| 159 |
print(f"Success in generating the paper.")
|
| 160 |
|
| 161 |
# Complete.
|