alexkueck commited on
Commit
df5275a
·
1 Parent(s): 03c2d80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -140,7 +140,7 @@ trainer = Trainer(
140
  )
141
  print ("trainer")
142
  #trainer ausführen
143
- trainer.train()
144
  #Wenn man vom letzten checkpoint aus weiter trainieren möchte: trainer.train(resume_from_checkpoint=True)
145
  print("trained!!!!!")
146
  #in den Hub laden
@@ -188,6 +188,7 @@ print("done")
188
  ##############################################
189
  #Testen des fine-tuned Modells
190
  print("Predict")
 
191
  model_neu = "test-tis-1"
192
  tokenizer_neu,model_neu, device_neu = load_tokenizer_and_model(model_neu, False)
193
  antwort = predict(model_neu, tokenizer_neu, device_neu, "Was ist Tis?", [["Tis", ""]], top_p=5,
 
140
  )
141
  print ("trainer")
142
  #trainer ausführen
143
+ #trainer.train()
144
  #Wenn man vom letzten checkpoint aus weiter trainieren möchte: trainer.train(resume_from_checkpoint=True)
145
  print("trained!!!!!")
146
  #in den Hub laden
 
188
  ##############################################
189
  #Testen des fine-tuned Modells
190
  print("Predict")
191
+ login(token=os.environ["HF_ACCESS_TOKEN"])
192
  model_neu = "test-tis-1"
193
  tokenizer_neu,model_neu, device_neu = load_tokenizer_and_model(model_neu, False)
194
  antwort = predict(model_neu, tokenizer_neu, device_neu, "Was ist Tis?", [["Tis", ""]], top_p=5,