karanthacker commited on
Commit
add4f31
1 Parent(s): 4285315

fix func args

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ model.load_state_dict(torch.load(hf_hub_download(repo_id="karanthacker/chat_ai",
33
  filename="alpaca_weights.pt"), map_location=device))
34
  model.eval()
35
 
36
- def respond(search_type, input):
37
  model.eval()
38
  src = torch.tensor(text_pipeline(input), dtype=torch.int64).unsqueeze(0).to(device)
39
  src_mask = ((src != pad_token) & (src != unknown_token)).unsqueeze(-2).to(device)
 
33
  filename="alpaca_weights.pt"), map_location=device))
34
  model.eval()
35
 
36
+ def respond(input):
37
  model.eval()
38
  src = torch.tensor(text_pipeline(input), dtype=torch.int64).unsqueeze(0).to(device)
39
  src_mask = ((src != pad_token) & (src != unknown_token)).unsqueeze(-2).to(device)