Cegil commited on
Commit
f7e25a6
1 Parent(s): 17951b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
  from transformers import T5ForConditionalGeneration, T5Tokenizer # Replace with the library used for your code generation model
3
-
4
  # Load your code generation model
5
  model_path = "Cegil/code_generation" # e.g., a model checkpoint or saved weights
6
  code_gen_model = T5ForConditionalGeneration.from_pretrained(model_path)
 
1
  import streamlit as st
2
  from transformers import T5ForConditionalGeneration, T5Tokenizer # Replace with the library used for your code generation model
3
+ import sentencepiece as spm
4
  # Load your code generation model
5
  model_path = "Cegil/code_generation" # e.g., a model checkpoint or saved weights
6
  code_gen_model = T5ForConditionalGeneration.from_pretrained(model_path)