alexkueck commited on
Commit
3e7d36b
·
1 Parent(s): 57e67fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -143,6 +143,11 @@ outputs = trainer.model(**encoding)
143
  logits = outputs.logits
144
  print(logits.shape)
145
 
 
 
 
 
 
146
 
147
  print("Save to Space")
148
  trainer.save_model("alexkueck/spaces/LIFineTuned/")
@@ -160,7 +165,10 @@ print("done")
160
  ##############################################
161
  #Testen des fine-tuned Modells
162
  print("Predict")
163
- antwort = predict("Was ist Tis?", [["Tis", None]])
 
 
 
164
  print(antwort)
165
  print("done Predict")
166
 
 
143
  logits = outputs.logits
144
  print(logits.shape)
145
 
146
+ #greedy_output = model.generate(input_ids, max_length=50)
147
+
148
+ print("Output:\n" )
149
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
150
+
151
 
152
  print("Save to Space")
153
  trainer.save_model("alexkueck/spaces/LIFineTuned/")
 
165
  ##############################################
166
  #Testen des fine-tuned Modells
167
  print("Predict")
168
+ antwort = predict("Was ist Tis?", [["Tis", None]], top_p=5,
169
+ temperature=0.8,
170
+ max_length_tokens=1024,
171
+ max_context_length_tokens=2048,)
172
  print(antwort)
173
  print("done Predict")
174