Spaces:

zjunlp
/

Chatcell

Running

sadgaj commited on Feb 4

Commit

d8d0f8b

•

1 Parent(s): c11da7b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,10 +8,10 @@ from transformers import (
     AutoModelForSeq2SeqLM,
 )
 import torch
-# model_folder = "zjunlp/chatcell-small"
-# tokenizer = AutoTokenizer.from_pretrained(model_folder)
-# model = AutoModelForSeq2SeqLM.from_pretrained(model_folder)
-# model.eval()

     AutoModelForSeq2SeqLM,
 )
 import torch
+model_folder = "zjunlp/chatcell-small"
+tokenizer = AutoTokenizer.from_pretrained(model_folder)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_folder)
+model.eval()