cahya commited on
Commit
28efd24
1 Parent(s): 80a49ac

fixed the result

Browse files
Files changed (1) hide show
  1. app.py +7 -4
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.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")