Update agent_t5.py

agent_t5.py

@@ -14,8 +14,6 @@ import os, time, torch
 from torch.nn import Softmax
 import requests
 
-API_URL = "https://api-inference.huggingface.co/models/CreatorPhan/ViQA-small"
-headers = {"Authorization": "Bearer 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"}
 
 class Agent:
     def __init__(self, args=None) -> None:
@@ -48,14 +46,6 @@ class Agent:
         self.use_context = True
 
         return f"Using file from {doc_path.name}"
-    
-    def API_call(self, prompt):
-        response = requests.post(API_URL, headers=headers, json={"inputs": prompt}).json()
-        if isinstance(response, list):
-            return response[0]['generated_text']
-        else:
-            time.sleep(3)
-            return self.API_call(prompt)
 
     def asking(self, question):
         timestamp = datetime.now()