Spaces:
Paused
Paused
Update app.py
Browse files
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,
|