Update app.py

app.py

@@ -28,12 +28,18 @@ class HoroscopeRetriever(RagRetriever):
         self.docs = docs
 
     def retrieve(self, question_hidden_states, n_docs=1):
-        # Convert the question_hidden_states (which is a numpy array) to text
-        question = question_hidden_states[0]
+        # Convert the question_hidden_states to a text string
+        question_hidden_states = question_hidden_states[0]
         
-        if isinstance(question, np.ndarray):
-            question = question.item()  # Convert the numpy array to a string
-        question = str(question).lower()
+        if isinstance(question_hidden_states, np.ndarray):
+            if question_hidden_states.size == 1:
+                question = question_hidden_states.item()  # Convert single-element array to scalar
+            else:
+                question = str(question_hidden_states[0])  # Take the first element of the array
+        else:
+            question = str(question_hidden_states)
+        
+        question = question.lower()
         
         # Simple retrieval logic: return the most relevant document based on the question
         for doc in self.docs: