Update app.py

app.py

@@ -26,11 +26,11 @@ def generate_story_from_caption(caption):
                         repo_id = "tiiuae/falcon-7b-instruct",
                         verbose = False,
                         model_kwargs = {"temperature":0.2, "max_new_tokens": 4000})
-    # template= """You are a story teller.
-    # You get a scenario as an input text, and generates a short story out of it.
-    # Context: {scenario}
-    # Story:
-    # """
+    template= """You are a story teller.
+    You get a scenario as an input text, and generates a short story out of it.
+    Context: {scenario}
+    Story:
+    """
     prompt = PromptTemplate(template=template, input_variables=["scenario"])
     #Let's create our LLM chain now
     chain = LLMChain(prompt=prompt, llm=llm)