Yin Fang
commited on
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;">
|