Spaces:

kodetr
/

stunting-llm

Sleeping

kodetr commited on Apr 22

Commit

39fb517

verified ·

1 Parent(s): 93403ab

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,6 +5,7 @@ import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 import os
 from threading import Thread
 HF_TOKEN = os.environ.get("HF_TOKEN", None)

 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 import os
 from threading import Thread
+from transformers import pipeline
 HF_TOKEN = os.environ.get("HF_TOKEN", None)