Abdul-Haseeb commited on
Commit
f45a3af
·
verified ·
1 Parent(s): e897e48

Update pipelines.py

Browse files
Files changed (1) hide show
  1. pipelines.py +2 -2
pipelines.py CHANGED
@@ -47,14 +47,14 @@ def generate_quiz_pipeline():
47
  quiz_generation_pipeline.add_component("link_content_fetcher", LinkContentFetcher())
48
  quiz_generation_pipeline.add_component("html_converter", HTMLToDocument())
49
  quiz_generation_pipeline.add_component("prompt_builder", ChatPromptBuilder(template=quiz_generation_template))
50
- quiz_generation_pipeline.add_component(
51
  "generator",
52
  HuggingFaceAPIGenerator(api_type=api_type,
53
  api_params={'model': 'meta-llama/Llama-3.1-8B-Instruct'
54
  },
55
  token=Secret.from_token(HUGGINGFACE_API_KEY),
56
  )
57
- )
58
  quiz_generation_pipeline.add_component(
59
  "generator",
60
  HuggingFaceAPIChatGenerator(
 
47
  quiz_generation_pipeline.add_component("link_content_fetcher", LinkContentFetcher())
48
  quiz_generation_pipeline.add_component("html_converter", HTMLToDocument())
49
  quiz_generation_pipeline.add_component("prompt_builder", ChatPromptBuilder(template=quiz_generation_template))
50
+ '''quiz_generation_pipeline.add_component(
51
  "generator",
52
  HuggingFaceAPIGenerator(api_type=api_type,
53
  api_params={'model': 'meta-llama/Llama-3.1-8B-Instruct'
54
  },
55
  token=Secret.from_token(HUGGINGFACE_API_KEY),
56
  )
57
+ )'''
58
  quiz_generation_pipeline.add_component(
59
  "generator",
60
  HuggingFaceAPIChatGenerator(