Spaces:

ChinarQ-AI
/

Image2Story

Running

Uzaiir commited on 5 days ago

Commit

b862bde

verified ·

1 Parent(s): 7353e33

Update src/models/storyGeneration.py (#8)

- Update src/models/storyGeneration.py (a9805010d27e75efc87a41015d8aede5d448cb5c)

Co-authored-by: Khan <[email protected]>

Files changed (1) hide show

src/models/storyGeneration.py CHANGED Viewed

@@ -4,7 +4,7 @@ import os
 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=200

 hf_token = os.environ.get("HUGGINGFACE_API_TOKEN")
 story_generator = HuggingFaceEndpoint(
+    repo_id="mistralai/Mistral-7B-Instruct-v0.3",
     huggingfacehub_api_token=hf_token,
     temperature=0.7,
     max_new_tokens=200