Spaces:

arinbalyan
/

codellm-python

Paused

arinbalyan commited on 25 days ago

Commit

25358c2

verified ·

1 Parent(s): ed865eb

Upload app.py with huggingface_hub

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-MODEL_ID = "arinbalyan/starcoder-code-lm"
 theme = (
     gr.themes.Soft(primary_hue="indigo", neutral_hue="slate")

 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+MODEL_ID = "arinbalyan/starcoder2-code-lm"
 theme = (
     gr.themes.Soft(primary_hue="indigo", neutral_hue="slate")