patpizio commited on
Commit
99c5f63
·
1 Parent(s): 5a1ea92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
- from transformers import AutoTokenizer, T5Tokenizer, T5ForConditionalGeneration, GenerationConfig, AutoModelForCausalLM
2
  import streamlit as st
 
3
 
4
  st.title('How does the LLM choose its words?')
5
 
 
1
+ import torch
2
  import streamlit as st
3
+ from transformers import AutoTokenizer, T5Tokenizer, T5ForConditionalGeneration, GenerationConfig, AutoModelForCausalLM
4
 
5
  st.title('How does the LLM choose its words?')
6