changed_api_key_2
Browse files- src/generator.py +1 -1
src/generator.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
|
4 |
class CustomGenerator:
|
5 |
def __init__(self, model_name='llama-3.3-70b-versatile',
|
6 |
-
api_key=os.getenv(api_key)):
|
7 |
self.model = model_name
|
8 |
self.client = Client(api_key=api_key)
|
9 |
self.system_prompt = """
|
|
|
3 |
|
4 |
class CustomGenerator:
|
5 |
def __init__(self, model_name='llama-3.3-70b-versatile',
|
6 |
+
api_key=os.getenv('api_key')):
|
7 |
self.model = model_name
|
8 |
self.client = Client(api_key=api_key)
|
9 |
self.system_prompt = """
|