alexkueck commited on
Commit
f73de62
·
1 Parent(s): fb17833

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -214,7 +214,7 @@ model_name_neu = "alexkueck/test-tis-1"
214
  #model_neu = trainer.load("test-tis-1")
215
 
216
  #oder ...
217
- model_neu, device_neu = load_tokenizer_and_model(model_name_neu, False)
218
  print("done load")
219
 
220
 
@@ -228,7 +228,7 @@ text = "Was ist Tis?"
228
  # Example usage
229
  prompt = "Was ist TIS?"
230
 
231
- response = generate_response(prompt, model_neu, tokenizer)
232
  print(response)
233
  print("response done")
234
 
 
214
  #model_neu = trainer.load("test-tis-1")
215
 
216
  #oder ...
217
+ model_neu, tokenizer_neu = load_tokenizer_and_model(model_name_neu, False)
218
  print("done load")
219
 
220
 
 
228
  # Example usage
229
  prompt = "Was ist TIS?"
230
 
231
+ response = generate_response(prompt, model_neu, tokenizer_neu)
232
  print(response)
233
  print("response done")
234