Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,8 @@ load_dotenv()
|
|
| 14 |
llm = HuggingFaceInferenceAPI(
|
| 15 |
# model_name="meta-llama/Meta-Llama-3-8B-Instruct",
|
| 16 |
model_name="mistralai/Mixtral-8x7B-Instruct-v0.1",
|
| 17 |
-
token=os.getenv("TOKEN")
|
|
|
|
| 18 |
)
|
| 19 |
|
| 20 |
parser = LlamaParse(api_key=os.getenv("LLAMA_INDEX_API"), result_type='markdown')
|
|
@@ -46,7 +47,7 @@ def respond(message, history):
|
|
| 46 |
# for chr in bot_message:
|
| 47 |
# output += chr
|
| 48 |
# yield output
|
| 49 |
-
print(f"\n{datetime.now()}::message --> {str(bot_message)}\n")
|
| 50 |
return str(bot_message)
|
| 51 |
except Exception as e:
|
| 52 |
if e == "'NoneType' object has no attribute 'as_query_engine'":
|
|
@@ -62,10 +63,10 @@ with gr.Blocks() as demo:
|
|
| 62 |
clear = gr.ClearButton()
|
| 63 |
btn = gr.Button("Submit", variant='primary')
|
| 64 |
output = gr.Text(label='Vector Index')
|
| 65 |
-
with gr.Column(scale=
|
| 66 |
gr.ChatInterface(fn=respond,
|
| 67 |
chatbot=gr.Chatbot(height=500),
|
| 68 |
-
textbox=gr.Textbox(placeholder="Ask me
|
| 69 |
# examples=["summarize the document"]
|
| 70 |
)
|
| 71 |
|
|
@@ -76,4 +77,4 @@ with gr.Blocks() as demo:
|
|
| 76 |
|
| 77 |
# Launch the demo with public link option
|
| 78 |
if __name__ == "__main__":
|
| 79 |
-
demo.launch(
|
|
|
|
| 14 |
llm = HuggingFaceInferenceAPI(
|
| 15 |
# model_name="meta-llama/Meta-Llama-3-8B-Instruct",
|
| 16 |
model_name="mistralai/Mixtral-8x7B-Instruct-v0.1",
|
| 17 |
+
token=os.getenv("TOKEN"),
|
| 18 |
+
model_config={'protected_namespaces': ()}
|
| 19 |
)
|
| 20 |
|
| 21 |
parser = LlamaParse(api_key=os.getenv("LLAMA_INDEX_API"), result_type='markdown')
|
|
|
|
| 47 |
# for chr in bot_message:
|
| 48 |
# output += chr
|
| 49 |
# yield output
|
| 50 |
+
print(f"\n{datetime.now()}:: {message} --> {str(bot_message)}\n")
|
| 51 |
return str(bot_message)
|
| 52 |
except Exception as e:
|
| 53 |
if e == "'NoneType' object has no attribute 'as_query_engine'":
|
|
|
|
| 63 |
clear = gr.ClearButton()
|
| 64 |
btn = gr.Button("Submit", variant='primary')
|
| 65 |
output = gr.Text(label='Vector Index')
|
| 66 |
+
with gr.Column(scale=3):
|
| 67 |
gr.ChatInterface(fn=respond,
|
| 68 |
chatbot=gr.Chatbot(height=500),
|
| 69 |
+
textbox=gr.Textbox(placeholder="Ask me questions on given document!", container=False, scale=7),
|
| 70 |
# examples=["summarize the document"]
|
| 71 |
)
|
| 72 |
|
|
|
|
| 77 |
|
| 78 |
# Launch the demo with public link option
|
| 79 |
if __name__ == "__main__":
|
| 80 |
+
demo.launch()
|