Akshay Vs commited on
Commit
0b540e1
1 Parent(s): ef5aca4

Added Models

Browse files
Files changed (1) hide show
  1. app.py +32 -2
app.py CHANGED
@@ -1,4 +1,34 @@
 
 
1
  import streamlit as st
 
 
 
2
 
3
- x = st.slider('Select a value')
4
- st.write(x, 'squared is', x * x)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ import time
3
  import streamlit as st
4
+ import concurrent.futures
5
+ from transformers import pipeline, set_seed
6
+ from random import randint
7
 
8
+ generator = pipeline('text-generation', model='openai-gpt')
9
+
10
+ def generate(initial_text, length=10, return_sequences=1):
11
+ set_seed(randint(1,1000))
12
+ result = generator(initial_text, max_length = length, num_return_sequences = return_sequences)
13
+ return result[0]["generated_text"]
14
+
15
+ def slice(text, mak_length=10):
16
+ return text[-mak_length:]
17
+
18
+ def type_text(text):
19
+ for letter in text:
20
+ sys.stdout.write(letter)
21
+ time.sleep(0)
22
+
23
+ if __name__ == "__main__":
24
+ text = input("Enter something to begin with... ")
25
+ print(".\n.\n.\nGenerating\n.\n.\n.")
26
+
27
+ for _ in range(50):
28
+ result = generate(text)
29
+ text=slice(result)
30
+ with concurrent.futures.ThreadPoolExecutor() as executor:
31
+ executor.submit(type_text, result.replace(text,""))
32
+
33
+ x = st.slider('Select a value')
34
+ st.write(x, 'squared is', x * x)