Spaces:

richardr1126
/

sql-skeleton-wizardcoder-demo

Paused

richardr1126 commited on Jul 21, 2023

Commit

f7fa740

•

1 Parent(s): cf0db0a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,13 +7,10 @@ from transformers import (
     StoppingCriteriaList,
     TextIteratorStreamer,
 )
-from huggingface_hub import login
 import gradio as gr
 import torch
-login(os.getenv("HF_TOKEN", None))
-model_name = "richardr1126/spider-skeleton-wizard-coder-merged"
 tok = AutoTokenizer.from_pretrained(model_name)
 max_new_tokens = 1536

     StoppingCriteriaList,
     TextIteratorStreamer,
 )
 import gradio as gr
 import torch
+model_name = os.getenv("HF_MODEL_NAME", None)
 tok = AutoTokenizer.from_pretrained(model_name)
 max_new_tokens = 1536