Spaces:

Tuana
/

find-the-animal

Runtime error

Tuana commited on Dec 15, 2022

Commit

a7a0ca8

1 Parent(s): 76fbd7f

changing the model

Files changed (1) hide show

utils/haystack.py CHANGED Viewed

@@ -47,7 +47,7 @@ def start_haystack():
         model_format="sentence_transformers",
     )
-    reader = FARMReader(model_name_or_path="deepset/roberta-base-squad2", use_gpu=True)
     retriever_images = MultiModalRetriever(

         model_format="sentence_transformers",
     )
+    reader = FARMReader(model_name_or_path="deepset/deberta-v3-base-squad2", use_gpu=True)
     retriever_images = MultiModalRetriever(