choco-conoz commited on
Commit
816c0dd
·
1 Parent(s): b191eec

feat: remove unused code

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -2
src/streamlit_app.py CHANGED
@@ -1,6 +1,5 @@
1
  import streamlit as st
2
- import torch
3
- from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer, pipeline
4
 
5
  # model_id = "sentence-transformers/all-MiniLM-L6-v2"
6
  # model_id = "sentence-transformers/xlm-r-base-en-ko-nli-ststb"
 
1
  import streamlit as st
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
3
 
4
  # model_id = "sentence-transformers/all-MiniLM-L6-v2"
5
  # model_id = "sentence-transformers/xlm-r-base-en-ko-nli-ststb"