k

app.py

@@ -85,16 +85,13 @@ def compute_similarity(text1, text2):
 def query_zephyr_model(text1, text2, question):
     prompt = f"Compare the following two contracts and answer the question:\nText 1: {text1}\nText 2: {text2}\nQuestion: {question}"
     try:
-        result = inference_client(inputs=prompt)
-        if 'generated_text' in result:
-            return result['generated_text']
-        else:
-            st.error("No generated text found in the response.")
-            return None
+        result = inference_client.text_generation(prompt)
+        return result.generated_text
     except Exception as e:
         st.error(f"Error querying the model: {e}")
         return None
 
+
 # ========== MAIN ========== 
 def main():
     st.title("📑 Contract Analyzer")