Update app.py

app.py

@@ -12,15 +12,6 @@ HF_TOKEN = os.getenv("HF_TOKEN", None)
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
-    def __init__(self):
-        print("BasicAgent initialized.")
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
-
 class BasicAgent:
     def __init__(self):
         # initialize HF inference pipeline once
@@ -30,6 +21,7 @@ class BasicAgent:
             "text-generation",
             model="gpt2",               # or your chosen model
             tokenizer="gpt2",
+            framework="pt",
             device=-1,                  # CPU; adjust if GPU
             use_auth_token=HF_TOKEN,
         )