Update app.py

app.py

@@ -123,6 +123,7 @@ training_args = TrainingArguments(
     #save_total_limit=1,
     #fp16=True,
     save_strategy = "no",
+    optim="adamw_torch"
     load_best_model_at_end=False,
     #push_to_hub=True,
 )
@@ -162,12 +163,12 @@ text = "Was ist Tis?"
 
 #outputs = trainer.model(**encoding)
 #logits = outputs.logits
-print(logits.shape)
+#print(logits.shape)
 
 #greedy_output = model.generate(input_ids, max_length=50)
 
 print("Output:\n" )
-print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+#print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
 
 print("Save to Space")