Spaces:
Runtime error
Runtime error
fixed the result
Browse files
app.py
CHANGED
@@ -18,6 +18,7 @@ MODELS = {
|
|
18 |
|
19 |
def query(payload, model_name):
|
20 |
data = json.dumps(payload)
|
|
|
21 |
response = requests.request("POST", MODELS[model_name]["url"], headers=headers, data=data)
|
22 |
return json.loads(response.content.decode("utf-8"))
|
23 |
|
@@ -59,7 +60,7 @@ max_len = st.sidebar.text_input(
|
|
59 |
temp = st.sidebar.slider(
|
60 |
"Temperature",
|
61 |
value=1.0,
|
62 |
-
min_value=0.
|
63 |
max_value=100.0,
|
64 |
help="The value used to module the next token probabilities."
|
65 |
)
|
@@ -97,12 +98,14 @@ text = st.text_area("Enter text", prompt_box)
|
|
97 |
if st.button("Run"):
|
98 |
with st.spinner(text="Getting results..."):
|
99 |
st.subheader("Result")
|
|
|
100 |
result = process(text=text,
|
101 |
model_name=model_name,
|
102 |
-
max_len=max_len,
|
103 |
temp=temp,
|
104 |
-
top_k=top_k,
|
105 |
-
top_p=top_p)
|
|
|
106 |
result = result[0]["generated_text"]
|
107 |
st.write(result.replace("\n", " \n"))
|
108 |
st.text("English translation")
|
|
|
18 |
|
19 |
def query(payload, model_name):
|
20 |
data = json.dumps(payload)
|
21 |
+
print("model url:", MODELS[model_name]["url"])
|
22 |
response = requests.request("POST", MODELS[model_name]["url"], headers=headers, data=data)
|
23 |
return json.loads(response.content.decode("utf-8"))
|
24 |
|
|
|
60 |
temp = st.sidebar.slider(
|
61 |
"Temperature",
|
62 |
value=1.0,
|
63 |
+
min_value=0.1,
|
64 |
max_value=100.0,
|
65 |
help="The value used to module the next token probabilities."
|
66 |
)
|
|
|
98 |
if st.button("Run"):
|
99 |
with st.spinner(text="Getting results..."):
|
100 |
st.subheader("Result")
|
101 |
+
print(f"maxlen:{max_len}, temp:{temp}, top_k:{top_k}, top_p:{top_p}")
|
102 |
result = process(text=text,
|
103 |
model_name=model_name,
|
104 |
+
max_len=int(max_len),
|
105 |
temp=temp,
|
106 |
+
top_k=int(top_k),
|
107 |
+
top_p=float(top_p))
|
108 |
+
print("result:", result)
|
109 |
result = result[0]["generated_text"]
|
110 |
st.write(result.replace("\n", " \n"))
|
111 |
st.text("English translation")
|