App

app.py

@@ -0,0 +1,34 @@
+from dotenv import load_dotenv
+from langchain import LLMChain, HuggingFaceHub
+from langchain.prompts import PromptTemplate
+import gradio as gr
+
+load_dotenv()
+
+
+def storyTime(input_letter):
+    hub_llm = HuggingFaceHub(
+        repo_id="gpt2-large",
+        model_kwargs={'temperature': 0.5, 'max_length': 200}
+    )
+    prompt = PromptTemplate(
+        input_variables=["letter"],
+        template=" {letter}"
+    )
+
+    hub_chain = LLMChain(prompt=prompt, llm=hub_llm, verbose=True)
+    return hub_chain.run(input_letter)
+
+
+iface = gr.Interface(
+    fn=storyTime,
+    inputs='text',
+    outputs=['text'],
+    title='Story Time😃',
+    description='Write a simple text you want a story on'
+)
+
+iface.launch(inline=False)
+
+# input_letter = input("Write the first sentence of your letter:  ")
+# main(input_letter)