bilgeyucel commited on
Commit
59b10b2
1 Parent(s): 95efa40

Add inference dependency

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -25,7 +25,6 @@ prompt_node = PromptNode(model_name_or_path="google/flan-t5-large", default_prom
25
  # prompt_node = PromptNode(model_name_or_path="tiiuae/falcon-7b-instruct", api_key=hf_api_key, default_prompt_template=pt, model_kwargs={"trust_remote_code":True})
26
 
27
  captioning_pipeline = Pipeline()
28
-
29
  captioning_pipeline.add_node(component=image_to_text, name="image_to_text", inputs=["File"])
30
  captioning_pipeline.add_node(component=prompt_node, name="prompt_node", inputs=["image_to_text"])
31
 
 
25
  # prompt_node = PromptNode(model_name_or_path="tiiuae/falcon-7b-instruct", api_key=hf_api_key, default_prompt_template=pt, model_kwargs={"trust_remote_code":True})
26
 
27
  captioning_pipeline = Pipeline()
 
28
  captioning_pipeline.add_node(component=image_to_text, name="image_to_text", inputs=["File"])
29
  captioning_pipeline.add_node(component=prompt_node, name="prompt_node", inputs=["image_to_text"])
30
 
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  farm-haystack==1.19.0
2
- gradio==3.39.0
 
 
1
  farm-haystack==1.19.0
2
+ gradio==3.39.0
3
+ farm-haystack[inference]