Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
-# Load CodeT5 model and tokenizer
-model_name = "Salesforce/codet5-base"
+model_name = "t5-small"  # Correct model for T5
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name)
 
@@ -12,7 +11,6 @@ def generate_comments(code):
     commented_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return commented_code
 
-# Create Gradio interface
 iface = gr.Interface(
     fn=generate_comments,
     inputs="text",