Hugging Face's logo

Spaces:
TimurHromek
/
HROM-V1
Running

App Files Community
4
TimurHromek commited on
Commit
76b9d13
·
verified ·
1 Parent(s): 5129b3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def load_model():
33
  return model
34
 
35
  model = load_model()
36
- safety = SafetyManager(model, tokenizer)
37
  max_response_length = 200
38
 
39
  def generate_response(model, tokenizer, input_ids, safety_manager, max_length=200):
 
33
  return model
34
 
35
  model = load_model()
36
+ safety = SafetyManager(model, tokenizer, device)
37
  max_response_length = 200
38
 
39
  def generate_response(model, tokenizer, input_ids, safety_manager, max_length=200):