aixsatoshi commited on
Commit
8830af9
1 Parent(s): 15181ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -13
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>elyza/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,17 +24,7 @@ Feel free to test without log.
24
  </center>
25
  """
26
 
27
- CSS = """
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(css=CSS) as demo:
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(