Akshay Vs
Added Models
0b540e1
raw
history blame
1.01 kB
import sys
import time
import streamlit as st
import concurrent.futures
from transformers import pipeline, set_seed
from random import randint
generator = pipeline('text-generation', model='openai-gpt')
def generate(initial_text, length=10, return_sequences=1):
set_seed(randint(1,1000))
result = generator(initial_text, max_length = length, num_return_sequences = return_sequences)
return result[0]["generated_text"]
def slice(text, mak_length=10):
return text[-mak_length:]
def type_text(text):
for letter in text:
sys.stdout.write(letter)
time.sleep(0)
if __name__ == "__main__":
text = input("Enter something to begin with... ")
print(".\n.\n.\nGenerating\n.\n.\n.")
for _ in range(50):
result = generate(text)
text=slice(result)
with concurrent.futures.ThreadPoolExecutor() as executor:
executor.submit(type_text, result.replace(text,""))
x = st.slider('Select a value')
st.write(x, 'squared is', x * x)