RuudVelo commited on
Commit
9299174
1 Parent(s): bceabb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(text, return_tensors="pt")
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)