Spaces:
Runtime error
Runtime error
karanthacker
commited on
Commit
·
add4f31
1
Parent(s):
4285315
fix func args
Browse files
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(
|
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)
|