Cegil commited on
Commit
fc83e8c
·
verified ·
1 Parent(s): 093b210

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import streamlit as st
2
- import T5ForConditionalGeneration # 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
 
1
  import streamlit as st
2
+ from transformers import T5ForConditionalGeneration # 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