Spaces:
Running
Running
Upload app.py
Browse files
app.py
CHANGED
|
@@ -106,7 +106,7 @@ def add_pdf_gradio(pdf_file_binary, progress=gr.Progress()):
|
|
| 106 |
|
| 107 |
return knowledge_base_name
|
| 108 |
|
| 109 |
-
|
| 110 |
def retrieve_gradio(knowledge_base: str, query: str, topk: int):
|
| 111 |
global model, tokenizer
|
| 112 |
|
|
|
|
| 106 |
|
| 107 |
return knowledge_base_name
|
| 108 |
|
| 109 |
+
@spaces.GPU
|
| 110 |
def retrieve_gradio(knowledge_base: str, query: str, topk: int):
|
| 111 |
global model, tokenizer
|
| 112 |
|