chaman420 commited on
Commit
d5c8765
·
verified ·
1 Parent(s): d62d1f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,5 +1,8 @@
1
  import streamlit as st
2
- from transformers import pipeline
 
 
 
3
 
4
 
5
  pipe = pipeline('sentiment-analysis')
 
1
  import streamlit as st
2
+ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
3
+
4
+ tokenizer = AutoTokenizer.from_pretrained("nakodanei/Blue-Orchid-2x7b")
5
+ model = AutoModelForCausalLM.from_pretrained("nakodanei/Blue-Orchid-2x7b")
6
 
7
 
8
  pipe = pipeline('sentiment-analysis')