nanoLLaVA

Running on Zero

qnguyen3 commited on Apr 8

Commit

90cce3a

•

1 Parent(s): 2002e2a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
 from threading import Thread
 import re
 import time

 import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 from threading import Thread
 import re
 import time