Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ input = st.text_input('Context')
|
|
55 |
|
56 |
if st.button('Submit'):
|
57 |
with st.spinner('Generating a response...'):
|
58 |
-
encoding = tokenizer(
|
59 |
outputs = model(**encoding)
|
60 |
predictions = outputs.logits.argmax(-1)
|
61 |
probabilities = torch.nn.functional.softmax(outputs.logits, dim=-1)
|
|
|
55 |
|
56 |
if st.button('Submit'):
|
57 |
with st.spinner('Generating a response...'):
|
58 |
+
encoding = tokenizer(input, return_tensors="pt")
|
59 |
outputs = model(**encoding)
|
60 |
predictions = outputs.logits.argmax(-1)
|
61 |
probabilities = torch.nn.functional.softmax(outputs.logits, dim=-1)
|