Flavio commited on
Commit
5998493
·
1 Parent(s): 24675b9
Files changed (1) hide show
  1. app.py +37 -2
app.py CHANGED
@@ -1,5 +1,40 @@
1
  import streamlit as st
 
 
2
 
3
- x = st.slider('Select a value')
4
- st.write(x, 'squared is', x * x)
5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
+ from transformers import pipeline
3
+ from HFBot import HFBot
4
 
 
 
5
 
6
+ def main():
7
+ modelliLLM = [
8
+ 'Almawave/Velvet-2B',
9
+ 'Almawave/Velvet-14B',
10
+ 'mistralai/Mistral-7B-Instruct-v0.1',
11
+ 'sapienzanlp/Minerva-350M-base-v1.0',
12
+ 'Qwen/Qwen3-0.6B'
13
+ ]
14
+ modello_scelto = st.selectbox("Seleziona un modello:", modelliLLM, index=0)
15
+ st.write(f"Hai selezionato: {modello_scelto}")
16
+
17
+ LLM = HFBot(model_name=modello_scelto)
18
+ domanda ="Valuta questo testo"
19
+ Istruzioni = "Rispondi sinteticamente"
20
+ frammenti = [
21
+ "visto il DPR 445/2000" , "con sui si elegge il presidente della repubblica"
22
+ ]
23
+
24
+ #res = LLM.chat(domanda=domanda, istruzioni= Istruzioni, frammenti=frammenti)
25
+ #st.write(f"Domanda {domanda} - risposta {res}")
26
+
27
+
28
+
29
+ if "atto_caricato" not in st.session_state:
30
+ st.subheader("Carica l'atto principale (Determinazione)")
31
+
32
+ main_pdf = st.file_uploader("Carica 1 file PDF - Determinazione", type=["pdf"], key="main_pdf")
33
+ if main_pdf:
34
+ st.session_state.main_pdf_nome = main_pdf.name
35
+ st.session_state["atto_caricato"] = True
36
+ else:
37
+ st.success(f'Caricato file {st.session_state.main_pdf_nome} ')
38
+
39
+ if __name__ == "__main__":
40
+ main()