Spaces:

boazchung
/

koalpaca-355m2

Sleeping

boazchung commited on Feb 15

Commit

27a7c55

•

1 Parent(s): 72a195b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,7 @@
-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"

 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"