Spaces:

TeamTonic
/

TonicsYI-6B-200k

Paused

Tonic commited on Nov 12, 2023

Commit

d43a99c

•

1 Parent(s): 71dee40

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,8 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from tokenization_yi import YiTokenizer
 import torch
 # Load the model and tokenizer
 model_name = "01-ai/Yi-34B-200K"

 from transformers import AutoModelForCausalLM, AutoTokenizer
 from tokenization_yi import YiTokenizer
 import torch
+import os
+import gradio as gr
 # Load the model and tokenizer
 model_name = "01-ai/Yi-34B-200K"