ZJU-Fangyin commited on
Commit
796b0f2
1 Parent(s): ceaf58c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import (
8
  AutoModelForSeq2SeqLM,
9
  )
10
  import torch
11
- model_folder = "zjunlp/chatcell-base"
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.baidu.com'>🐣Project</a> &nbsp;&nbsp;&nbsp; <a href='https://www.baidu.com'>📃Paper</a>&nbsp;&nbsp;&nbsp; <a href='https://www.baidu.com'>🥳Code</a></p>")
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> &nbsp;&nbsp;&nbsp; <a href='https://arxiv.org/abs/2306.08018'>📃Paper</a>&nbsp;&nbsp;&nbsp; <a href='https://github.com/zjunlp/ChatCell'>🥳Code</a></p>")
71
 
72
  # gr.HTML("""
73
  # <div style="text-align: center; margin: 0 auto;">