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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -174,7 +174,7 @@ print("done")
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,
 
174
  ##############################################
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,