Update app.py

app.py

@@ -1,9 +1,10 @@
 import streamlit as st
-from transformers import T5ForConditionalGeneration  # Replace with the library used for your code generation model
+from transformers import T5ForConditionalGeneration, T5Tokenizer  # Replace with the library used for your code generation model
 
 # Load your code generation model
 model_path = "Cegil/code_generation"  # e.g., a model checkpoint or saved weights
-code_gen_model = T5ForConditionalGeneration.load_model(model_path)  # Load your model
+code_gen_model = T5ForConditionalGeneration.from_pretrained(model_path)
+tokenizer = T5Tokenizer.from_pretrained(model_path)
 
 # UI for Code Generation
 st.title("Code Generation Interface")