Update src/models/storyGeneration.py

src/models/storyGeneration.py

@@ -1,16 +1,16 @@
-# from langchain_huggingface import HuggingFaceEndpoint
-# import os
+from langchain_huggingface import HuggingFaceEndpoint
+import os
 
-# hf_token = os.environ.get("HUGGINGFACE_API_TOKEN")
+hf_token = os.environ.get("HUGGINGFACE_API_TOKEN")
 
-# story_generator = HuggingFaceEndpoint(
-#     repo_id="HuggingFaceH4/zephyr-7b-beta",
-#     huggingfacehub_api_token=hf_token,
-#     temperature=0.7,
-#     max_new_tokens=800
-# )
+story_generator = HuggingFaceEndpoint(
+    repo_id="HuggingFaceH4/zephyr-7b-beta",
+    huggingfacehub_api_token=hf_token,
+    temperature=0.7,
+    max_new_tokens=800
+)
 
-# def generateStory(prompt):
-#     if not prompt:
-#         return "Invalid caption input."
-#     return story_generator.invoke(prompt)
+def generateStory(prompt):
+    if not prompt:
+        return "Invalid caption input."
+    return story_generator.invoke(prompt)