Cegil commited on
Commit
130d8d3
·
verified ·
1 Parent(s): 6f0f9e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import streamlit as st
2
- import some_code_generation_library # 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 = some_code_generation_library.load_model(model_path) # Load your model
7
 
8
  # UI for Code Generation
9
  st.title("Code Generation Interface")
 
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
6
+ code_gen_model = T5ForConditionalGeneration.load_model(model_path) # Load your model
7
 
8
  # UI for Code Generation
9
  st.title("Code Generation Interface")