Update app.py

app.py

@@ -1,11 +1,13 @@
 import streamlit as st
 import os
+import requests
+
 
 #from transformers import BertModel, BertTokenizer
 from transformers import HfAgent, load_tool
 
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, LocalAgent
+from transformers import AutoModelForCausalLM, AutoTokenizer, Agent, LocalAgent
 
 
 #checkpoint = "THUDM/agentlm-7b"
@@ -180,7 +182,7 @@ def handle_submission():
     # Initialize the agent with the selected tools
     #agent = HfAgent("https://api-inference.huggingface.co/models/bigcode/starcoder", additional_tools=tools)
     #agent = HfAgent("https://api-inference.huggingface.co/models/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", additional_tools=tools)
-    agent = HfAgent("https://api-inference.huggingface.co/models/THUDM/agentlm-7b", additional_tools=tools)
+    #agent = HfAgent("https://api-inference.huggingface.co/models/THUDM/agentlm-7b", additional_tools=tools)
     
 
  #   agent.config.tokenizer = tokenizer