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)