alexkueck commited on
Commit
21a4e7c
·
1 Parent(s): 03d727b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -194,7 +194,7 @@ print("done")
194
  #Testen des fine-tuned Modells
195
  print("Predict")
196
  login(token=os.environ["HF_ACCESS_TOKEN"]) #read access token!!!!!!!!!!
197
- model_name = "test-tis-1"
198
  tokenizer_neu = load_tokenizer(base_model)
199
  model_neu, device_neu = load_model(model_name, False)
200
  antwort = predict(model_neu, tokenizer_neu, device_neu, "Was ist Tis?", [["Tis", ""]], top_p=5,
 
194
  #Testen des fine-tuned Modells
195
  print("Predict")
196
  login(token=os.environ["HF_ACCESS_TOKEN"]) #read access token!!!!!!!!!!
197
+ model_name = "alexkueck/test-tis-1"
198
  tokenizer_neu = load_tokenizer(base_model)
199
  model_neu, device_neu = load_model(model_name, False)
200
  antwort = predict(model_neu, tokenizer_neu, device_neu, "Was ist Tis?", [["Tis", ""]], top_p=5,