ZJU-Fangyin
commited on
Commit
•
796b0f2
1
Parent(s):
ceaf58c
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from transformers import (
|
|
8 |
AutoModelForSeq2SeqLM,
|
9 |
)
|
10 |
import torch
|
11 |
-
model_folder = "zjunlp/chatcell-
|
12 |
tokenizer = AutoTokenizer.from_pretrained(model_folder)
|
13 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_folder)
|
14 |
if torch.cuda.is_available():
|
@@ -67,7 +67,7 @@ with gr.Blocks(css=css,
|
|
67 |
""")
|
68 |
# gr.HTML("")
|
69 |
with gr.Row():
|
70 |
-
gr.Markdown("<p align='center'><a href='https://www.
|
71 |
|
72 |
# gr.HTML("""
|
73 |
# <div style="text-align: center; margin: 0 auto;">
|
|
|
8 |
AutoModelForSeq2SeqLM,
|
9 |
)
|
10 |
import torch
|
11 |
+
model_folder = "zjunlp/chatcell-large"
|
12 |
tokenizer = AutoTokenizer.from_pretrained(model_folder)
|
13 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_folder)
|
14 |
if torch.cuda.is_available():
|
|
|
67 |
""")
|
68 |
# gr.HTML("")
|
69 |
with gr.Row():
|
70 |
+
gr.Markdown("<p align='center'><a href='https://www.zjukg.org/project/ChatCell'>🐣Project</a> <a href='https://arxiv.org/abs/2306.08018'>📃Paper</a> <a href='https://github.com/zjunlp/ChatCell'>🥳Code</a></p>")
|
71 |
|
72 |
# gr.HTML("""
|
73 |
# <div style="text-align: center; margin: 0 auto;">
|