Skip to content

Commit 8565189

Browse files
authored
Update chat.py
1 parent 8e857f0 commit 8565189

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

chatbot/chat.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,9 +38,7 @@
3838

3939
output = model(X)
4040
_, predicted = torch.max(output, dim=1)
41-
print(predicted)
4241
tag = tags[predicted.item()]
43-
print(tag)
4442

4543
probs = torch.softmax(output, dim=1)
4644
prob = probs[0][predicted.item()]

0 commit comments

Comments
 (0)