Spaces:

boazchung
/

koalpaca-355m2

Sleeping

boazchung commited on Feb 15

Commit

72a195b

•

1 Parent(s): 25d62a9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
-import transformers
 import gradio as gr
 def get_pipe():
-    from transformers import AutoTokenizer, AutoModelForCausalLM
     model_name = "heegyu/koalpaca-355m"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     tokenizer.truncation_side = "right"

+from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 def get_pipe():
     model_name = "heegyu/koalpaca-355m"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     tokenizer.truncation_side = "right"