Add first attempt files

README.md

@@ -1,6 +1,6 @@
 ---
 title: SonIA
-emoji: 🏃
+emoji: 🤖
 colorFrom: indigo
 colorTo: blue
 sdk: streamlit

app.py

@@ -0,0 +1,52 @@
+import streamlit as st
+from transformers import Pipeline, pipeline
+import tokenizers
+import torch
+import json
+
+@st.cache(
+    hash_funcs={
+        torch.nn.parameter.Parameter: lambda _: None,
+        tokenizers.Tokenizer: lambda _: None,
+        tokenizers.AddedToken: lambda _: None,
+    },
+    allow_output_mutation=True,
+    show_spinner=False,
+)
+
+def load_engine() -> Pipeline:
+
+    nlp_qa = pipeline(
+        'question-answering',
+        model='mrm8488/bert-italian-finedtuned-squadv1-it-alfa',
+        tokenizer='mrm8488/bert-italian-finedtuned-squadv1-it-alfa'
+    )
+
+    return nlp_qa
+
+with st.spinner(text="Sto preparando il necessario per rispondere alle tue domande personali..."):
+
+    engine = load_engine()
+
+with open("knowledge.json") as f:
+    context = json.load(f)
+
+st.title("La macchina che risponde alle tue domande personali")
+
+input = st.text_input("Fai una domanda e comparirà la risposta!")
+
+if input:
+    try:
+
+        answer = engine(
+            {
+                'question': input,
+                'context': context["info"]
+            }
+        )
+
+        st.subheader(answer["answer"])
+
+    except:
+
+        st.error("Qualcosa é andato storto. Prova di nuovo con un'altra domanda magari!")

context.json

@@ -0,0 +1,2 @@
+{"info": "Sei Giuseppe, hai 69 anni e vivi a Villafranca di Verona in Corso Vittorio Emanuele 238. Ora siamo a novembre dell'anno 2021 e hai avuto un'encefalite virale a gennaio 2021. Vivi assieme a Davide. Davide è tuo figlio e ha 37 anni. A causa dell'encefalite ora soffri di amnesia. Quandi hai perso la memoria. Tua moglie Raffaella vive a Verona. Raffaella sta bene. Con Raffaella hai rapporti cordiali. I tuoi fratelli Alessandro, Giovanni e Grazia stanno bene."
+}