Spaces:
Sleeping
Sleeping
Zai
commited on
Commit
·
8a0563d
1
Parent(s):
2d880e2
fixed interface.py
Browse files- interface.py +20 -18
interface.py
CHANGED
@@ -1,13 +1,14 @@
|
|
1 |
import streamlit as st
|
|
|
|
|
2 |
|
3 |
|
4 |
def main():
|
5 |
-
st.set_page_config(page_title="
|
6 |
|
7 |
-
st.title("
|
8 |
-
st.write("Welcome to our Dream
|
9 |
|
10 |
-
# Add a sidebar with options
|
11 |
st.sidebar.header("Select a Dream Theme")
|
12 |
theme = st.sidebar.selectbox(
|
13 |
"Choose a theme", ["Fantasy", "Adventure", "Romance", "Horror"]
|
@@ -16,32 +17,33 @@ def main():
|
|
16 |
st.sidebar.header("Customize Your Dream")
|
17 |
prompt = st.sidebar.text_input("Enter a prompt or keyword")
|
18 |
num_sentences = st.sidebar.slider("Number of sentences to generate", 1, 10, 5)
|
19 |
-
temperature = st.sidebar.slider("Creativity of LLM (Temperature)", 1,
|
20 |
|
21 |
generate_button = st.sidebar.button("Generate Dream")
|
22 |
|
23 |
-
# Create a main area to display the generated text
|
24 |
st.header("Your Dream")
|
25 |
|
26 |
if generate_button:
|
27 |
-
|
28 |
with st.spinner("Generating dream..."):
|
29 |
-
# Generate text using the LLM
|
30 |
-
# (assuming you have an LLM model implemented)
|
31 |
-
dream_text = generate_dream_text(theme, prompt, num_sentences)
|
32 |
|
33 |
-
|
|
|
34 |
st.text_area("", value=dream_text, height=600)
|
35 |
else:
|
36 |
-
text_area =
|
37 |
-
|
38 |
-
|
|
|
|
|
|
|
|
|
|
|
39 |
|
|
|
40 |
|
41 |
-
|
42 |
-
|
43 |
-
# For demonstration purposes, return a dummy text
|
44 |
-
return "This is a dummy dream text. Please implement your LLM model to generate actual text."
|
45 |
|
46 |
|
47 |
if __name__ == "__main__":
|
|
|
1 |
import streamlit as st
|
2 |
+
from yume import Yume
|
3 |
+
from yume.config import yume_small
|
4 |
|
5 |
|
6 |
def main():
|
7 |
+
st.set_page_config(page_title="夢の生成器", page_icon="🌙", layout="wide")
|
8 |
|
9 |
+
st.title("夢の生成器")
|
10 |
+
st.write("Welcome to our Dream Generator! 🌙")
|
11 |
|
|
|
12 |
st.sidebar.header("Select a Dream Theme")
|
13 |
theme = st.sidebar.selectbox(
|
14 |
"Choose a theme", ["Fantasy", "Adventure", "Romance", "Horror"]
|
|
|
17 |
st.sidebar.header("Customize Your Dream")
|
18 |
prompt = st.sidebar.text_input("Enter a prompt or keyword")
|
19 |
num_sentences = st.sidebar.slider("Number of sentences to generate", 1, 10, 5)
|
20 |
+
temperature = st.sidebar.slider("Creativity of LLM (Temperature)", 0.1, 1.0, 0.5)
|
21 |
|
22 |
generate_button = st.sidebar.button("Generate Dream")
|
23 |
|
|
|
24 |
st.header("Your Dream")
|
25 |
|
26 |
if generate_button:
|
27 |
+
|
28 |
with st.spinner("Generating dream..."):
|
|
|
|
|
|
|
29 |
|
30 |
+
dream_text = generate_dream_text(theme, prompt, num_sentences, temperature)
|
31 |
+
|
32 |
st.text_area("", value=dream_text, height=600)
|
33 |
else:
|
34 |
+
st.text_area("", height=600, placeholder="Your dream will appear here...")
|
35 |
+
|
36 |
+
|
37 |
+
def generate_dream_text(theme, prompt, num_sentences, temperature):
|
38 |
+
|
39 |
+
yume = Yume(config=yume_small)
|
40 |
+
|
41 |
+
yume.load_pretrained("zaibutcooler/yume")
|
42 |
|
43 |
+
full_prompt = f"{theme} - {prompt}"
|
44 |
|
45 |
+
# Generate text using the Yume model
|
46 |
+
return yume.sample(full_prompt)
|
|
|
|
|
47 |
|
48 |
|
49 |
if __name__ == "__main__":
|