Update app.py

app.py

@@ -159,7 +159,7 @@ class model:
         else: return inference_loss
 
     
-    def T2Isearch(self, query, focussed_word, i, k=50):
+    def T2Isearch(self, query, focussed_word, k=50):
         # Encode the text query        
         inputs = text_tokenizer([query, focussed_word], padding=True, return_tensors="pt")    
         outputs = text_model(**inputs)
@@ -221,4 +221,4 @@ Focussed_word = st.text_input("Enter your focussed word here:")
 if st.button("Search"):
     LCM = model(d, "pascal")
     if query:
-        LCM.T2Isearch(query, Focussed_word, int(ind))
+        LCM.T2Isearch(query, Focussed_word)