Spaces:

ckdeveloper
/

llm

Sleeping

ckdeveloper commited on Aug 21

Commit

4ff5dbc

verified ·

1 Parent(s): ac36394

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-model_id = "deepseek-ai/deepseek-coder-1.3b-base"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)

 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+model_id = "Qwen/Qwen2.5-Coder-7B"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)