Spaces:

Uzaiir
/

Image2Story

Sleeping

Uzaiir commited on Jul 25, 2025

Commit

69847e0

verified ·

1 Parent(s): c8bb585

Update src/models/storyGeneration.py

Files changed (1) hide show

src/models/storyGeneration.py CHANGED Viewed

@@ -8,6 +8,7 @@ story_generator = HuggingFaceEndpoint(
     huggingfacehub_api_token=hf_token,
     temperature=0.7,
     max_new_tokens=800
 )
 def generateStory(prompt):

     huggingfacehub_api_token=hf_token,
     temperature=0.7,
     max_new_tokens=800
+    inference_provider="auto"
 )
 def generateStory(prompt):