Jimin Park commited on
Commit
0854291
·
1 Parent(s): 014e235

updated req.txt and app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  from transformers import pipeline
 
2
  from peft import PeftModel, PeftConfig
3
  import gradio as gr
4
 
 
1
  from transformers import pipeline
2
+ from transformers import AutoModelForCausalLM, AutoTokenizer
3
  from peft import PeftModel, PeftConfig
4
  import gradio as gr
5