Spaces:

Ritesh-hf
/

Nutrigenics-chatbot

Sleeping

Ritesh-hf commited on Oct 20, 2024

Commit

99490d9

verified ·

1 Parent(s): f36e45e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -162,7 +162,7 @@ class Chat:
     def get_target(self, img_feats, tar_img_feats) :
         score = (img_feats @ tar_img_feats.t()).squeeze(0).cpu().detach().numpy()
-        index = np.argsort(score)[::-1][0] + 1
         self.target_recipe = df.iloc[index]
     def ask(self):

     def get_target(self, img_feats, tar_img_feats) :
         score = (img_feats @ tar_img_feats.t()).squeeze(0).cpu().detach().numpy()
+        index = np.argsort(score)[::-1][0]
         self.target_recipe = df.iloc[index]
     def ask(self):