changed model codellama to code t5

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 import faiss
 import os
@@ -14,11 +14,12 @@ if hf_token:
 else:
     st.error("Token not found")
     st.stop()
-    
-# 🎯 Load CodeLlama 3B Model
-model_name = "codellama/CodeLlama-3b"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto",use_auth_token=hf_token)
+
+
+# Load code T5 model 
+tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")
+model = AutoModelForSeq2SeqLM.from_pretrained("Salesforce/codet5-base")
+
 
 # 🔍 Function to Generate Responses
 def generate_response(prompt):