didivarian commited on
Commit
776ad9a
·
verified ·
1 Parent(s): 7df1a57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,3 +1,5 @@
1
- import gradio as gr
 
2
 
3
- gr.load("models/Bin12345/AutoCoder").launch()
 
 
1
+ # Load model directly
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
+ tokenizer = AutoTokenizer.from_pretrained("Bin12345/AutoCoder")
5
+ model = AutoModelForCausalLM.from_pretrained("Bin12345/AutoCoder")