Spaces:
Sleeping
Sleeping
Upload 2 files
Browse files- a.py +11 -0
- requirements.txt +13 -0
a.py
ADDED
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
from transformers import pipeline
|
3 |
+
|
4 |
+
pipe = pipeline(
|
5 |
+
task='text-generation',
|
6 |
+
model = 'meta-llama/Meta-Llama-3-8B'
|
7 |
+
)
|
8 |
+
|
9 |
+
x = st.chat_input(placeholder='Your Message')
|
10 |
+
o = pipe("Hi")
|
11 |
+
st.write(o[0]['generated_text'])
|
requirements.txt
ADDED
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
langchain==0.0.184
|
2 |
+
PyPDF2==3.0.1
|
3 |
+
python-dotenv==1.0.0
|
4 |
+
streamlit==1.18.1
|
5 |
+
faiss-cpu==1.7.4
|
6 |
+
altair==4
|
7 |
+
tiktoken==0.4.0
|
8 |
+
# uncomment to use huggingface llms
|
9 |
+
huggingface-hub==0.14.1
|
10 |
+
|
11 |
+
# uncomment to use instructor embeddings
|
12 |
+
InstructorEmbedding==1.0.1
|
13 |
+
sentence-transformers==2.2.2
|