Silas-Blanchard commited on
Commit
03954af
·
verified ·
1 Parent(s): 914e42c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -1,4 +1,11 @@
1
  import streamlit as st
2
 
3
  x = st.slider('Select a value')
4
- st.write(x, 'squared is', x * x)
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
 
3
  x = st.slider('Select a value')
4
+ st.write(x, 'squared is', x * x)
5
+
6
+ from transformers import pipeline
7
+ generator2 = pipeline('text-generation', model='meta-llama/Llama-2-7b-chat-hf')
8
+
9
+ prompt = "Return an opening move in chess."
10
+ generated_text = generator2(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
11
+ print(generated_text)