Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ MODEL_ID = "elyza/Llama-3-ELYZA-JP-8B"
|
|
| 12 |
MODELS = os.environ.get("MODELS")
|
| 13 |
MODEL_NAME = MODELS.split("/")[-1]
|
| 14 |
|
| 15 |
-
TITLE = "<h1><center>
|
| 16 |
|
| 17 |
DESCRIPTION = f"""
|
| 18 |
<h3>MODEL: <a href="https://hf.co/{MODELS}">{MODEL_NAME}</a></h3>
|
|
@@ -24,17 +24,7 @@ Feel free to test without log.
|
|
| 24 |
</center>
|
| 25 |
"""
|
| 26 |
|
| 27 |
-
|
| 28 |
-
.duplicate-button {
|
| 29 |
-
margin: auto !important;
|
| 30 |
-
color: white !important;
|
| 31 |
-
background: black !important;
|
| 32 |
-
border-radius: 100vh !important;
|
| 33 |
-
}
|
| 34 |
-
h3 {
|
| 35 |
-
text-align: center;
|
| 36 |
-
}
|
| 37 |
-
"""
|
| 38 |
|
| 39 |
model = AutoModelForCausalLM.from_pretrained(
|
| 40 |
MODELS,
|
|
@@ -83,7 +73,7 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
|
|
| 83 |
|
| 84 |
chatbot = gr.Chatbot(height=500)
|
| 85 |
|
| 86 |
-
with gr.Blocks(
|
| 87 |
gr.HTML(TITLE)
|
| 88 |
gr.HTML(DESCRIPTION)
|
| 89 |
gr.ChatInterface(
|
|
|
|
| 12 |
MODELS = os.environ.get("MODELS")
|
| 13 |
MODEL_NAME = MODELS.split("/")[-1]
|
| 14 |
|
| 15 |
+
TITLE = "<h1><center>Llama-3-ELYZA-JP-8B Chat webui</center></h1>"
|
| 16 |
|
| 17 |
DESCRIPTION = f"""
|
| 18 |
<h3>MODEL: <a href="https://hf.co/{MODELS}">{MODEL_NAME}</a></h3>
|
|
|
|
| 24 |
</center>
|
| 25 |
"""
|
| 26 |
|
| 27 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
model = AutoModelForCausalLM.from_pretrained(
|
| 30 |
MODELS,
|
|
|
|
| 73 |
|
| 74 |
chatbot = gr.Chatbot(height=500)
|
| 75 |
|
| 76 |
+
with gr.Blocks() as demo:
|
| 77 |
gr.HTML(TITLE)
|
| 78 |
gr.HTML(DESCRIPTION)
|
| 79 |
gr.ChatInterface(
|