aixsatoshi
commited on
Commit
•
8830af9
1
Parent(s):
15181ca
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(
|