Update app.py

app.py

@@ -1,23 +1,18 @@
 import requests
 import streamlit as st
 import streamlit.components.v1 as components
-
+def query(payload):
+    	response = requests.post(API_URL, headers=headers, json=payload)
+    	return response.json()
+    
 def semanticComparativeClassification():
     st.write(st.session_state.mytext)
     
     #API_URL = "https://api-inference.huggingface.co/models/Maite89/Roberta_finetuning_semantic_similarity_stsb_multi_mt"
     API_URL = "https://api-inference.huggingface.co/models/symanto/sn-xlm-roberta-base-snli-mnli-anli-xnli"
     headers = {"Authorization": "Bearer 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"}
-    
-    def query(payload):
-    	response = requests.post(API_URL, headers=headers, json=payload)
-    	return response.json()
-    	
-    output = query({
-    	"inputs": {
-            "wait_for_model" : True,
-    		"source_sentence": st.session_state.mytext,
-    		"sentences": [
+
+    sentences =  [
     			"Conoces Lya2",
     			"He perdido la contraseña",
     			"Calendario de eventos",
@@ -26,6 +21,12 @@ def semanticComparativeClassification():
                 "Hola me estás saludando",
                 "Adiós te despides"
     		]
+     	
+    output = query({
+    	"inputs": {
+            "wait_for_model" : True,
+    		"source_sentence": st.session_state.mytext,
+    		"sentences": sentences
     	},
     })
     index = 0
@@ -37,7 +38,7 @@ def semanticComparativeClassification():
             sentenceindex = index
         index = index + 1
 
-    st.write("Hablamos de ", str(sentenceindex)) 
+    st.write("Hablamos de ", sentences[sentenceindex]) 
     
     return output