Spaces:

vilarin
/

vllm-chat

Paused

vilarin commited on Jun 7

Commit

e476f12

•

1 Parent(s): 4e2b6c5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,10 +5,10 @@ from transformers import AutoTokenizer
 from vllm import LLM, SamplingParams
 model = os.environ.get("MODEL_ID")
-MODEL_NAME = model.split("/")[-1]
 DESCRIPTION = f"""
-<h3>MODEL: <a href="https://hf.co/{MODELS}">{MODEL_NAME}</a></h3>
 <center>
 <p>Qwen is the large language model built by Alibaba Cloud.
 <br>

 from vllm import LLM, SamplingParams
 model = os.environ.get("MODEL_ID")
+model_name = model.split("/")[-1]
 DESCRIPTION = f"""
+<h3>MODEL: <a href="https://hf.co/{model}">{model_name}</a></h3>
 <center>
 <p>Qwen is the large language model built by Alibaba Cloud.
 <br>