Spaces:

avciTheProgrammer
/

code-generator

Sleeping

avciTheProgrammer commited on Jul 12

Commit

a4e0a36

verified ·

1 Parent(s): 32b1a25

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,10 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 # Set model ID
-model_id = "deepseek-ai/deepseek-coder-1.3b-base"
 # model_id = "gpt2"
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(model_id)

 import torch
 # Set model ID
+# model_id = "deepseek-ai/deepseek-coder-1.3b-base"
 # model_id = "gpt2"
+model_id = "deepseek-ai/deepseek-coder-1.3b-instruct"
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(model_id)