alexkueck commited on
Commit
0589738
·
1 Parent(s): d3286bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -175,8 +175,8 @@ print("done")
175
  #Testen des fine-tuned Modells
176
  print("Predict")
177
  model_neu = "EleutherAI/gpt-neo-1.3B"
178
- tokenizer,model = load_tokenizer_and_model(model_neu)
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,)
 
175
  #Testen des fine-tuned Modells
176
  print("Predict")
177
  model_neu = "EleutherAI/gpt-neo-1.3B"
178
+ tokenizer_neu,model_neu, device_neu = load_tokenizer_and_model(model_neu, False)
179
+ antwort = predict(model_neu, tokenizer_neu, "Was ist Tis?", [["Tis", None]], top_p=5,
180
  temperature=0.8,
181
  max_length_tokens=1024,
182
  max_context_length_tokens=2048,)