Update app.py

app.py

@@ -1,9 +1,8 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-
-model_name = "Salesforce/codegen-350M-mono"
-base_model = AutoModelForCausalLM.from_pretrained(model_name, quantization_config=bnb_config, use_cache = False, device_map=device_map)
+model_name = "vijjuk/codegen-350M-mono-python-18k-alpaca"
+base_model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = "right"