Funybubble commited on
Commit
ffb307e
·
verified ·
1 Parent(s): d3f153c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -1,16 +1,13 @@
1
- from transformers import AutoModel, AutoTokenizer
2
 
3
  # Replace 'Funybubble/replit-code-instruct-quanatized' with the actual model name
4
  model_name = 'Funybubble/replit-code-instruct-quanatized'
5
 
6
  # Load the tokenizer and model
7
- tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModel.from_pretrained(model_name)
9
 
10
  # Save the tokenizer and model to the same directory as app.py
11
  save_directory = './'
12
- tokenizer.save_pretrained(save_directory)
13
- model.save_pretrained(save_directory)
14
 
15
- # Rest of your code using the tokenizer and model
16
- # ...
 
1
+ from transformers import AutoModel
2
 
3
  # Replace 'Funybubble/replit-code-instruct-quanatized' with the actual model name
4
  model_name = 'Funybubble/replit-code-instruct-quanatized'
5
 
6
  # Load the tokenizer and model
7
+
8
  model = AutoModel.from_pretrained(model_name)
9
 
10
  # Save the tokenizer and model to the same directory as app.py
11
  save_directory = './'
 
 
12
 
13
+ model.save_pretrained(save_directory)