alexkueck commited on
Commit
b6ee899
·
1 Parent(s): 8716861

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -174,7 +174,9 @@ print("done")
174
  ##############################################
175
  #Testen des fine-tuned Modells
176
  print("Predict")
177
- antwort = predict("Was ist Tis?", [["Tis", None]], top_p=5,
 
 
178
  temperature=0.8,
179
  max_length_tokens=1024,
180
  max_context_length_tokens=2048,)
 
174
  ##############################################
175
  #Testen des fine-tuned Modells
176
  print("Predict")
177
+ model_neu = "xxx"
178
+ tokenizer,model,device = load_tokenizer_and_model(model_neu,False)
179
+ antwort = predict(model, tokenizer, "Was ist Tis?", [["Tis", None]], top_p=5,
180
  temperature=0.8,
181
  max_length_tokens=1024,
182
  max_context_length_tokens=2048,)