lukmanaj commited on
Commit
5245852
·
verified ·
1 Parent(s): 8c10ba1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ class GeminiModel:
43
  frequency_penalty=0.0,
44
  )
45
 
46
- def __call__(self, prompt: str) -> str:
47
  """Send prompt to Gemini."""
48
  try:
49
  response = self.client.generate_content(
 
43
  frequency_penalty=0.0,
44
  )
45
 
46
+ def __call__(self, prompt: str, **kwargs) -> str:
47
  """Send prompt to Gemini."""
48
  try:
49
  response = self.client.generate_content(