Zai commited on
Commit
8a0563d
·
1 Parent(s): 2d880e2

fixed interface.py

Browse files
Files changed (1) hide show
  1. 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="夢を見ている", page_icon="🌙", layout="wide")
6
 
7
- st.title("夢を見ている")
8
- st.write("Welcome to our Dream Writer! 🌙")
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, 10, 5)
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
- # Add a loading animation
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
- # Display the generated text
 
34
  st.text_area("", value=dream_text, height=600)
35
  else:
36
- text_area = st.text_area(
37
- "", height=600, placeholder="Your dream will appear here..."
38
- )
 
 
 
 
 
39
 
 
40
 
41
- def generate_dream_text(theme, prompt, num_sentences):
42
- # Implement your LLM model here to generate text
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__":