Update app.py

app.py

@@ -7,6 +7,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
 MAX_MAX_NEW_TOKENS = 1024
 MAX_INPUT_TOKEN_LENGTH = 2048
 
+# base_model_name = "m-a-p/OpenCodeInterpreter-DS-6.7B"
 base_model_name = "m-a-p/OpenCodeInterpreter-DS-1.3B"
 model = AutoModelForCausalLM.from_pretrained(base_model_name, torch_dtype=torch.float32, device_map="cpu", low_cpu_mem_usage=True)