Spaces:
Sleeping
Sleeping
Update utils.py
Browse files
utils.py
CHANGED
@@ -23,7 +23,7 @@ def generate_script(prompt,video_length,creativity,api_key):
|
|
23 |
'temperature': creativity
|
24 |
}
|
25 |
|
26 |
-
llm = HuggingFaceHub(repo_id ="
|
27 |
|
28 |
#Creating chain for 'Title' & 'Video Script'
|
29 |
title_chain = LLMChain(llm=llm, prompt=title_template, verbose=True)
|
|
|
23 |
'temperature': creativity
|
24 |
}
|
25 |
|
26 |
+
llm = HuggingFaceHub(repo_id ="distilbert/distilgpt2", model_kwargs = model_kwargs)
|
27 |
|
28 |
#Creating chain for 'Title' & 'Video Script'
|
29 |
title_chain = LLMChain(llm=llm, prompt=title_template, verbose=True)
|