Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -130,7 +130,7 @@ Infer the date from context. Q: Today is the second day of the third month of 1
130
  sh = gc.open_by_url(gsheet_url)
131
 
132
  ws = sh.sheet1
133
- ws.insert_rows([[LORA_HUB_NAMES[i] for i in st.session_state["select_names"]],recommendation.value.tolist(),[]])
134
  st.table(df)
135
  random_id = uuid.uuid4().hex
136
  os.makedirs(f"lora/{random_id}")
@@ -149,13 +149,12 @@ Infer the date from context. Q: Today is the second day of the third month of 1
149
  )
150
  with open(f"lora_{random_id}.zip", "rb") as fp:
151
  btn = st.download_button(
152
- label="📥 Download the final LoRA Module and share your results",
153
  data=fp,
154
  file_name=f"lora_{random_id}.zip",
155
  mime="application/zip",
156
  on_click=share
157
  )
158
- st.button("📥 Share your results",on_click=share)
159
  st.warning("The page will be refreshed once you click the download button. Share results may cost 1-2 mins.")
160
 
161
 
 
130
  sh = gc.open_by_url(gsheet_url)
131
 
132
  ws = sh.sheet1
133
+ ws.insert_rows([[LORA_HUB_NAMES[i] for i in st.session_state["select_names"]],recommendation.value.tolist(),[max_step]])
134
  st.table(df)
135
  random_id = uuid.uuid4().hex
136
  os.makedirs(f"lora/{random_id}")
 
149
  )
150
  with open(f"lora_{random_id}.zip", "rb") as fp:
151
  btn = st.download_button(
152
+ label="📥 Download and share your results",
153
  data=fp,
154
  file_name=f"lora_{random_id}.zip",
155
  mime="application/zip",
156
  on_click=share
157
  )
 
158
  st.warning("The page will be refreshed once you click the download button. Share results may cost 1-2 mins.")
159
 
160