Update app.py

app.py

@@ -6,7 +6,7 @@ from accelerate import init_empty_weights, load_checkpoint_and_dispatch
 # Load the model using Accelerate for memory optimization
 @st.cache_resource()
 def load_model():
-    MODEL_NAME = "huggingface/CodeGen-2B"  # Updated model name
+    MODEL_NAME = "Salesforce/codegen-2B-multi"  # Updated model name
     tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
     
     # Load model with accelerate to optimize for memory usage