Spaces:

ChinarQ-AI
/

Image2Story

Running

Uzaiir commited on 7 days ago

Commit

a980501

verified ·

1 Parent(s): 7353e33

Update src/models/storyGeneration.py

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