JamalAG commited on
Commit
5ef08b3
1 Parent(s): 59ba65e

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
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 ="meta-llama/Llama-2-7b", model_kwargs = model_kwargs)
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)