suwonpabby commited on
Commit
171268d
1 Parent(s): dfe5753
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ for module in modules:
31
  print(f"{module} is not installed")
32
 
33
  #fmt: off
34
- from transformers import AutoModelForCausalLM, AutoTokenizer
35
  from FlagEmbedding import BGEM3FlagModel
36
 
37
 
 
31
  print(f"{module} is not installed")
32
 
33
  #fmt: off
34
+ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
35
  from FlagEmbedding import BGEM3FlagModel
36
 
37