Spaces:

patpizio
/

llm-token-probs

Sleeping

patpizio commited on Jul 9, 2023

Commit

99c5f63

1 Parent(s): 5a1ea92

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,6 @@
-from transformers import AutoTokenizer, T5Tokenizer, T5ForConditionalGeneration, GenerationConfig, AutoModelForCausalLM
 import streamlit as st
 st.title('How does the LLM choose its words?')

+import torch
 import streamlit as st
+from transformers import AutoTokenizer, T5Tokenizer, T5ForConditionalGeneration, GenerationConfig, AutoModelForCausalLM
 st.title('How does the LLM choose its words?')