Commit
•
3928b10
1
Parent(s):
3164600
Update app.py
Browse files
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 |
|