Spaces:
Runtime error
Runtime error
Benjamin Bossan
commited on
Commit
·
af69d1f
1
Parent(s):
1bd69d6
Fix issues with requirements
Browse files
app.py
CHANGED
@@ -5,7 +5,6 @@ from __future__ import annotations
|
|
5 |
import base64
|
6 |
import glob
|
7 |
import io
|
8 |
-
import json
|
9 |
import os
|
10 |
import pickle
|
11 |
import re
|
@@ -32,7 +31,11 @@ PLOT_PREFIX = "__plot__:"
|
|
32 |
if "custom_sections" not in st.session_state:
|
33 |
st.session_state.custom_sections = {}
|
34 |
|
|
|
35 |
tmp_repo = Path(mkdtemp(prefix="skops-"))
|
|
|
|
|
|
|
36 |
left_col, right_col = st.columns([1, 2])
|
37 |
|
38 |
# a hacky way to "persist" custom sections
|
@@ -61,17 +64,19 @@ def _write_plot(plot_name, plot_file):
|
|
61 |
|
62 |
|
63 |
def init_repo():
|
64 |
-
_clear_repo(
|
65 |
|
66 |
try:
|
67 |
-
file_name =
|
68 |
sio.dump(model, file_name)
|
|
|
|
|
69 |
hub_utils.init(
|
70 |
model=file_name,
|
71 |
-
dst=
|
72 |
task=task,
|
73 |
data=data,
|
74 |
-
requirements=
|
75 |
)
|
76 |
except Exception as exc:
|
77 |
print("Uh oh, something went wrong when initializing the repo:", exc)
|
@@ -116,7 +121,7 @@ def _create_model_card():
|
|
116 |
return
|
117 |
|
118 |
init_repo()
|
119 |
-
metadata = card.metadata_from_config(
|
120 |
model_card = card.Card(model=model, metadata=metadata)
|
121 |
|
122 |
if model_description:
|
@@ -283,9 +288,9 @@ with left_col:
|
|
283 |
on_change=init_repo,
|
284 |
)
|
285 |
|
286 |
-
requirements = st.
|
287 |
label="Requirements*",
|
288 |
-
value=
|
289 |
on_change=init_repo,
|
290 |
)
|
291 |
|
|
|
5 |
import base64
|
6 |
import glob
|
7 |
import io
|
|
|
8 |
import os
|
9 |
import pickle
|
10 |
import re
|
|
|
31 |
if "custom_sections" not in st.session_state:
|
32 |
st.session_state.custom_sections = {}
|
33 |
|
34 |
+
# the tmp_repo is used to upload the sklearn model to
|
35 |
tmp_repo = Path(mkdtemp(prefix="skops-"))
|
36 |
+
# the hf_repo is the actual repo used for init()
|
37 |
+
hf_repo = Path(mkdtemp(prefix="skops-"))
|
38 |
+
|
39 |
left_col, right_col = st.columns([1, 2])
|
40 |
|
41 |
# a hacky way to "persist" custom sections
|
|
|
64 |
|
65 |
|
66 |
def init_repo():
|
67 |
+
_clear_repo(hf_repo)
|
68 |
|
69 |
try:
|
70 |
+
file_name = tmp_repo / "model.skops"
|
71 |
sio.dump(model, file_name)
|
72 |
+
|
73 |
+
reqs = [r.strip().rstrip(",") for r in requirements.splitlines()]
|
74 |
hub_utils.init(
|
75 |
model=file_name,
|
76 |
+
dst=hf_repo,
|
77 |
task=task,
|
78 |
data=data,
|
79 |
+
requirements=reqs,
|
80 |
)
|
81 |
except Exception as exc:
|
82 |
print("Uh oh, something went wrong when initializing the repo:", exc)
|
|
|
121 |
return
|
122 |
|
123 |
init_repo()
|
124 |
+
metadata = card.metadata_from_config(hf_repo)
|
125 |
model_card = card.Card(model=model, metadata=metadata)
|
126 |
|
127 |
if model_description:
|
|
|
288 |
on_change=init_repo,
|
289 |
)
|
290 |
|
291 |
+
requirements = st.text_area(
|
292 |
label="Requirements*",
|
293 |
+
value=f"scikit-learn=={sklearn.__version__}\n",
|
294 |
on_change=init_repo,
|
295 |
)
|
296 |
|