awacke1 commited on
Commit
f4bf684
·
1 Parent(s): df65edd

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import requests
2
+ import streamlit as st
3
+
4
+ st.title("AI Story Generator")
5
+
6
+ # Input form for user to enter story details
7
+ story = st.text_area("Enter the beginning of your story:")
8
+ length = st.slider("Select the desired length of the story:", min_value=50, max_value=500, value=150, step=10)
9
+ temperature = st.slider("Select the temperature for the generated text:", min_value=0.1, max_value=2.0, value=0.7, step=0.1)
10
+ top_k = st.slider("Select the top-k value for the generated text:", min_value=1, max_value=50, value=1, step=1)
11
+ top_p = st.slider("Select the top-p value for the generated text:", min_value=0.1, max_value=1.0, value=0.2, step=0.1)
12
+
13
+ # Generate story based on user input
14
+ if st.button("Generate Story"):
15
+ response = requests.post("https://blinkdl-chatrwkv-gradio.hf.space/run/predict", json={
16
+ "data": [
17
+ story,
18
+ length,
19
+ top_k,
20
+ temperature,
21
+ top_p
22
+ ]
23
+ }).json()
24
+
25
+ # Display generated story to user
26
+ generated_text = response["data"]
27
+ st.write("Generated Story:")
28
+ st.write(generated_text)