Spaces:
Running
Running
Adrian Palma commited on
Commit ·
f0ff05a
1
Parent(s): 7a6c314
initial commit hf space
Browse files- tests/test_api.py +13 -0
tests/test_api.py
CHANGED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from gradio_client import Client
|
| 2 |
+
|
| 3 |
+
client = Client("https://adrianpalmadev-rag-qa-system.hf.space")
|
| 4 |
+
|
| 5 |
+
result = client.predict(
|
| 6 |
+
query="Where is the hospital?",
|
| 7 |
+
top_k=2,
|
| 8 |
+
umbral=0.5,
|
| 9 |
+
api_name="/ask"
|
| 10 |
+
)
|
| 11 |
+
|
| 12 |
+
print("Respuesta:")
|
| 13 |
+
print(result)
|