Final_Assignment_Project

Sleeping

wt002 commited on about 1 month ago

Commit

b9deb15

verified ·

1 Parent(s): 44c22e4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -182,8 +182,8 @@ class VideoTranscriptionTool(Tool):
 class BasicAgent:
     def __init__(self):
         token = os.environ.get("HF_API_TOKEN")
-        model = HfApiModel(
-            model_name="google/gemini-1.5-flash",
             temperature=0.1,
             token=token
         )

 class BasicAgent:
     def __init__(self):
         token = os.environ.get("HF_API_TOKEN")
+        self.model = HfApiModel(
+            repo_id="google/gemini-1.5-flash",
             temperature=0.1,
             token=token
         )