Spaces:
Running
Running
bilgeyucel
commited on
Commit
·
59b10b2
1
Parent(s):
95efa40
Add inference dependency
Browse files- app.py +0 -1
- 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]
|