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