Spaces:

Uzaiir
/

Image2Story

Sleeping

Uzaiir commited on Jul 25, 2025

Commit

29134b1

verified ·

1 Parent(s): 69847e0

Update src/models/storyGeneration.py

Files changed (1) hide show

src/models/storyGeneration.py CHANGED Viewed

@@ -7,7 +7,7 @@ story_generator = HuggingFaceEndpoint(
     repo_id="HuggingFaceH4/zephyr-7b-beta",
     huggingfacehub_api_token=hf_token,
     temperature=0.7,
-    max_new_tokens=800
     inference_provider="auto"
 )

     repo_id="HuggingFaceH4/zephyr-7b-beta",
     huggingfacehub_api_token=hf_token,
     temperature=0.7,
+    max_new_tokens=300,
     inference_provider="auto"
 )