Spaces:
Running
on
Zero
Running
on
Zero
zamalali
commited on
Commit
·
ae13177
1
Parent(s):
f1e023c
Enable GPU wrapper for repository ranking in DeepGit Lite
Browse files
app.py
CHANGED
|
@@ -102,7 +102,7 @@ def parse_result_to_html(raw_result: str) -> str:
|
|
| 102 |
# ---------------------------
|
| 103 |
# GPU-enabled Wrapper for Repository Ranking
|
| 104 |
# ---------------------------
|
| 105 |
-
|
| 106 |
def gpu_run_repo(topic: str):
|
| 107 |
return run_repository_ranking(topic)
|
| 108 |
|
|
|
|
| 102 |
# ---------------------------
|
| 103 |
# GPU-enabled Wrapper for Repository Ranking
|
| 104 |
# ---------------------------
|
| 105 |
+
@spaces.GPU(duration=180)
|
| 106 |
def gpu_run_repo(topic: str):
|
| 107 |
return run_repository_ranking(topic)
|
| 108 |
|