Included venv

app.py

@@ -17,8 +17,8 @@ rag = RAG(
     hf_token=os.getenv("HF_TOKEN"),
     embeddings_model=os.getenv("EMBEDDINGS"), 
     model_name=os.getenv("MODEL"),   
-    
-
+    rerank_model=os.getenv("RERANK_MODEL"),
+    rerank_number_contexts=int(os.getenv("RERANK_NUMBER_CONTEXTS"))
 )
 
 
@@ -35,6 +35,7 @@ def generate(prompt, model_parameters):
         gr.Warning(
             "Inference endpoint is not available right now. Please try again later."
         )
+    return None, None, None
 
 
 def submit_input(input_, num_chunks, max_new_tokens, repetition_penalty, top_k, top_p, do_sample, temperature):
@@ -58,8 +59,9 @@ def submit_input(input_, num_chunks, max_new_tokens, repetition_penalty, top_k,
 
     for url in source:
         sources_markup += f'<a href="{url}" target="_blank">{url}</a><br>'
-        
-    return output.strip(), sources_markup, context
+
+    return output, sources_markup, context  
+    # return output.strip(), sources_markup, context
 
 
 def change_interactive(text):