Update app.py

app.py

@@ -56,21 +56,6 @@ GEN_NAMES = [
     "huggingface/EleutherAI/gpt-j-6B",
     "huggingface/gpt2-large"
 ]
-GENERATORS = {}
-manager = ModelManager()
-for MODEL_NAME in GEN_NAMES :
-    print(f"downloading {MODEL_NAME}")
-    model_path, config_path, model_item = manager.download_model(f"{MODEL_NAME}")
-    #vocoder_name: Optional[str] = model_item["default_vocoder"]
-    #vocoder_path = None
-    #vocoder_config_path = None
-    #if vocoder_name is not None:
-    #    vocoder_path, vocoder_config_path, _ = manager.download_model(vocoder_name)
-
-    #synthesizer = Synthesizer(
-    #     model_path, config_path, None, vocoder_path, vocoder_config_path,
-    #)
-    #MODELS[MODEL_NAME] = synthesizer
 
 
 #ASR
@@ -81,6 +66,10 @@ def transcribe(audio):
 #Sentiment Classifier
 classifier = pipeline("text-classification")
 
+# GPT-J: Story Generation Pipeline
+story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
+
+
 #STT
 def speech_to_text(speech):
     text = asr(speech)["text"]
@@ -91,6 +80,11 @@ def text_to_sentiment(text):
     sentiment = classifier(text)[0]["label"]
     return sentiment 
 
+#TTStory
+def text_to_story(text):
+    story= story_gen(text)[0]["label"]
+    return story
+
 #Save
 def upsert(text):
     date_time =str(datetime.datetime.today())
@@ -164,7 +158,7 @@ with demo:
     b3.click(upsert, inputs=text, outputs=saved)
     b4.click(selectall, inputs=text, outputs=savedAll)
     b5.click(tts,  inputs=[text,TTSchoice], outputs=audio)
-    b6.click(ttstory,  inputs=[text,Storychoice], outputs=text)
+    b6.click(text_to_story,  inputs=[text,Storychoice], outputs=text)
 
 # Lets Do It
 demo.launch(share=True)
@@ -180,9 +174,4 @@ examples = [
     ["Collections of people have agency and mass having agency at the mesoscopic level"],
     ["Having a deep human connection is an interface problem to solve."],
     ["Having a collective creates agency since we build trust in eachother."]
-]
-
-#gr.Parallel(generator1, generator2, generator3, inputs=gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence."),
-#            title=title, examples=examples).launch(share=False)
-            
-            
+]