bwilkie commited on
Commit
9f08c4f
·
verified ·
1 Parent(s): ed9acbe

Update myagent.py

Browse files
Files changed (1) hide show
  1. myagent.py +2 -1
myagent.py CHANGED
@@ -4,7 +4,8 @@ from smolagents import OpenAIServerModel
4
  from tools.fetch import fetch_webpage
5
  from tools.yttranscript import get_youtube_transcript, get_youtube_title_description
6
  import myprompts
7
-
 
8
  # --- Basic Agent Definition ---
9
  class BasicAgent:
10
  def __init__(self):
 
4
  from tools.fetch import fetch_webpage
5
  from tools.yttranscript import get_youtube_transcript, get_youtube_title_description
6
  import myprompts
7
+ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
8
+ import torch
9
  # --- Basic Agent Definition ---
10
  class BasicAgent:
11
  def __init__(self):