justinj92 commited on
Commit
ec71f48
·
verified ·
1 Parent(s): 213c70e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from transformers import (
4
  AutoModelForCausalLM,
5
  AutoTokenizer,
6
  TextIteratorStreamer,
 
7
  )
8
  import os
9
  from threading import Thread
 
4
  AutoModelForCausalLM,
5
  AutoTokenizer,
6
  TextIteratorStreamer,
7
+ pipeline
8
  )
9
  import os
10
  from threading import Thread