tomas.helmfridsson commited on
Commit
b93e9df
·
1 Parent(s): df94e26

update guis 3

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -24,7 +24,7 @@ def load_vectorstore():
24
 
25
  with gr.Blocks() as demo:
26
  with gr.Row():
27
- status = 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,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()