Spaces:
Sleeping
Sleeping
tomas.helmfridsson
commited on
Commit
·
b93e9df
1
Parent(s):
df94e26
update guis 3
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def load_vectorstore():
|
|
24 |
|
25 |
with gr.Blocks() as demo:
|
26 |
with gr.Row():
|
27 |
-
|
28 |
|
29 |
vectorstore, loaded_files = load_vectorstore()
|
30 |
llm_pipeline = pipeline("text-generation", model="tiiuae/falcon-rw-1b", device=-1)
|
@@ -32,16 +32,17 @@ with gr.Blocks() as demo:
|
|
32 |
qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=vectorstore.as_retriever())
|
33 |
|
34 |
loaded_list = "\n".join([f"- {f}" for f in loaded_files])
|
|
|
35 |
gr.Markdown(f"✅ Klar! Du kan nu ställa frågor om dokumenten nedan:\n\n{loaded_list}")
|
36 |
|
37 |
def chat_fn(message, history):
|
38 |
if len(message) > 1000:
|
39 |
-
return "⚠️ Din fråga är för lång (" + str(len(message)) + " tecken). Försök ställa en mer specifik fråga."
|
40 |
try:
|
41 |
svar = qa_chain.invoke({"query": message})
|
42 |
except Exception as e:
|
43 |
-
return f"Ett fel uppstod vid bearbetning av frågan: {str(e)}"
|
44 |
-
return svar
|
45 |
|
46 |
gr.ChatInterface(
|
47 |
fn=chat_fn,
|
@@ -51,4 +52,3 @@ with gr.Blocks() as demo:
|
|
51 |
)
|
52 |
|
53 |
demo.launch()
|
54 |
-
|
|
|
24 |
|
25 |
with gr.Blocks() as demo:
|
26 |
with gr.Row():
|
27 |
+
status_text = gr.Markdown("🔄 Laddar modellen, vänta...")
|
28 |
|
29 |
vectorstore, loaded_files = load_vectorstore()
|
30 |
llm_pipeline = pipeline("text-generation", model="tiiuae/falcon-rw-1b", device=-1)
|
|
|
32 |
qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=vectorstore.as_retriever())
|
33 |
|
34 |
loaded_list = "\n".join([f"- {f}" for f in loaded_files])
|
35 |
+
status_text.visible = False
|
36 |
gr.Markdown(f"✅ Klar! Du kan nu ställa frågor om dokumenten nedan:\n\n{loaded_list}")
|
37 |
|
38 |
def chat_fn(message, history):
|
39 |
if len(message) > 1000:
|
40 |
+
return ("⚠️ Din fråga är för lång (" + str(len(message)) + " tecken). Försök ställa en mer specifik fråga.",)
|
41 |
try:
|
42 |
svar = qa_chain.invoke({"query": message})
|
43 |
except Exception as e:
|
44 |
+
return (f"Ett fel uppstod vid bearbetning av frågan: {str(e)}",)
|
45 |
+
return (svar["result"],)
|
46 |
|
47 |
gr.ChatInterface(
|
48 |
fn=chat_fn,
|
|
|
52 |
)
|
53 |
|
54 |
demo.launch()
|
|