Spaces:
Runtime error
Runtime error
minor fixes
Browse files- src/main.py +5 -5
src/main.py
CHANGED
@@ -117,7 +117,7 @@ def chunk_dataset(ds_name):
|
|
117 |
)
|
118 |
|
119 |
dataset.push_to_hub(
|
120 |
-
|
121 |
private=chunk_config.private,
|
122 |
token=HF_TOKEN
|
123 |
)
|
@@ -190,7 +190,7 @@ def embed_dataset(ds_name):
|
|
190 |
|
191 |
dataset = Dataset.from_json(temp_file.name)
|
192 |
dataset.push_to_hub(
|
193 |
-
|
194 |
private=embed_config.private,
|
195 |
token=HF_TOKEN
|
196 |
)
|
@@ -201,6 +201,6 @@ def embed_dataset(ds_name):
|
|
201 |
|
202 |
# For debugging
|
203 |
|
204 |
-
import uvicorn
|
205 |
-
if __name__ == "__main__":
|
206 |
-
|
|
|
117 |
)
|
118 |
|
119 |
dataset.push_to_hub(
|
120 |
+
CHUNKED_DS_NAME,
|
121 |
private=chunk_config.private,
|
122 |
token=HF_TOKEN
|
123 |
)
|
|
|
190 |
|
191 |
dataset = Dataset.from_json(temp_file.name)
|
192 |
dataset.push_to_hub(
|
193 |
+
EMBED_DS_NAME,
|
194 |
private=embed_config.private,
|
195 |
token=HF_TOKEN
|
196 |
)
|
|
|
201 |
|
202 |
# For debugging
|
203 |
|
204 |
+
# import uvicorn
|
205 |
+
# if __name__ == "__main__":
|
206 |
+
# uvicorn.run(app, host="0.0.0.0", port=7860)
|