alexkueck commited on
Commit
4d81104
·
1 Parent(s): 67bd156

Update app.py

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