multimodalart HF staff commited on
Commit
2834c0b
1 Parent(s): 0fd53b3

Add token to create discussion

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -266,7 +266,7 @@ Sample pictures of:
266
  extra_message = "Don't forget to remove the GPU attribution after you play with it."
267
  else:
268
  extra_message = "The GPU has been removed automatically as requested, and you can try the model via the model page"
269
- 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}")
270
 
271
  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"])]
272
 
 
266
  extra_message = "Don't forget to remove the GPU attribution after you play with it."
267
  else:
268
  extra_message = "The GPU has been removed automatically as requested, and you can try the model via the model page"
269
+ 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)
270
 
271
  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"])]
272