multimodalart HF staff commited on
Commit
3928b10
1 Parent(s): 3164600

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -283,7 +283,7 @@ Sample pictures of:
283
  extra_message = "Don't forget to remove the GPU attribution after you play with it."
284
  else:
285
  extra_message = "The GPU has been removed automatically as requested, and you can try the model via the model page"
286
- api.create_discussion(repo_id=os.environ['SPACE_ID'], title=f"Your model {model_name} has finished trained from the Dreambooth Train Spaces!", description=f"Your model has been successfully uploaded to: https://huggingface.co/{model_id}. {extra_message}",token=hf_token)
287
 
288
  return [gr.update(visible=True, value=f"Successfully uploaded your model. Access it [here](https://huggingface.co/{model_id})"), gr.update(visible=True, value=["diffusers_model.zip", "model.ckpt"])]
289
 
 
283
  extra_message = "Don't forget to remove the GPU attribution after you play with it."
284
  else:
285
  extra_message = "The GPU has been removed automatically as requested, and you can try the model via the model page"
286
+ api.create_discussion(repo_id=os.environ['SPACE_ID'], title=f"Your model {model_name} has finished trained from the Dreambooth Train Spaces!", description=f"Your model has been successfully uploaded to: https://huggingface.co/{model_id}. {extra_message}",repo_type="space", token=hf_token)
287
 
288
  return [gr.update(visible=True, value=f"Successfully uploaded your model. Access it [here](https://huggingface.co/{model_id})"), gr.update(visible=True, value=["diffusers_model.zip", "model.ckpt"])]
289