Spaces:

avciTheProgrammer
/

code-generator

Running

avciTheProgrammer commited on 6 days ago

Commit

32b1a25

verified ·

1 Parent(s): 31e1742

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ 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"
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(model_id)