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")
|