From 340b2ab70d9cb9906972a3673f339c26010bcbb3 Mon Sep 17 00:00:00 2001
From: osanseviero <osanseviero@gmail.com>
Date: Mon, 31 May 2021 11:44:00 +0200
Subject: [PATCH] SpaCy fixes

---
 interfaces/Libraries.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/interfaces/Libraries.ts b/interfaces/Libraries.ts
index 4ecc0baec3..81ab9bd004 100644
--- a/interfaces/Libraries.ts
+++ b/interfaces/Libraries.ts
@@ -162,7 +162,7 @@ const sentenceTransformers = (model: ModelData) =>
 model = SentenceTransformer("${model.modelId}")`;
 
 const spacy = (model: ModelData) =>
-`# pip install https://huggingface.co/${model.modelId}/resolve/main/${nameWithoutNamespace(model.modelId)}-any-py3-none-any.whl
+`!pip install https://huggingface.co/${model.modelId}/resolve/main/${nameWithoutNamespace(model.modelId)}-any-py3-none-any.whl
 
 # Importing as module.
 import ${nameWithoutNamespace(model.modelId)}
@@ -170,7 +170,7 @@ nlp = ${nameWithoutNamespace(model.modelId)}.load()
 
 # Using spaCy.load().
 import spacy
-nlp = spacy.load(${nameWithoutNamespace(model.modelId)})`;
+nlp = spacy.load("${nameWithoutNamespace(model.modelId)}")`;
 
 const transformers = (model: ModelData) =>
 `from transformers import AutoTokenizer, ${model.autoArchitecture}