alexkueck commited on
Commit
520185d
·
1 Parent(s): d7abeb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ print("trained!!!!!")
139
  #trainer.push_to_hub()
140
 
141
  print("Save to Space")
142
- trainer.save_model()
143
 
144
 
145
 
 
139
  #trainer.push_to_hub()
140
 
141
  print("Save to Space")
142
+ trainer.save_model("./my_model")
143
 
144
 
145