avciTheProgrammer commited on
Commit
e24ba6d
·
verified ·
1 Parent(s): 23baa65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ import torch
4
 
5
  # Set model ID
6
  # model_id = "deepseek-ai/deepseek-coder-1.3b-base"
7
- # model_id = "gpt2"
8
- model_id = "deepseek-ai/deepseek-coder-1.3b-instruct"
9
 
10
 
11
  # Load tokenizer and model
 
4
 
5
  # Set model ID
6
  # model_id = "deepseek-ai/deepseek-coder-1.3b-base"
7
+ model_id = "gpt2"
8
+ # model_id = "deepseek-ai/deepseek-coder-1.3b-instruct"
9
 
10
 
11
  # Load tokenizer and model