Update app.py

app.py

@@ -1,16 +1,14 @@
-from __future__ import annotations
-from typing import List, Dict, Union
-import os
-import re
-import subprocess
-import streamlit as st
-import time
 from langchain_core.prompts import PromptTemplate
+from langchain_core.runnables import RunnableSequence
+from huggingface_hub import HuggingFaceHub, InferenceApi as InferenceClient
 from langchain_community.llms import HuggingFaceEndpoint
-from huggingface_hub.inference_api import InferenceApi as InferenceClient
+from streamlit import StreamlitApp, write, text_input, text_area, button, session_state, write as st_write
+import os
+import time
 
 # Load LLM
-llm = HuggingFaceHub(repo_id="tiiuae/falcon-7b-instruct", model_kwargs={"temperature": 0.1, "max_new_tokens": 500})
+llm = HuggingFaceEndpoint(repo_id="tiiuae/falcon-7b-instruct", model_kwargs={"temperature": 0.1, "max_new_tokens": 500})
+
 
 class Agent:
     def __init__(self, name: str, agent_type: str, complexity: int):