Spaces:
Running
Running
File size: 311 Bytes
1c57e18 | 1 2 3 4 5 6 7 8 9 10 11 | from google import genai
class GeminiDriver:
def __init__(self):
self.client = genai.Client()
def generate_text(self, prompt: str) -> str:
response = self.client.models.generate_content(
model="gemini-3-flash-preview", contents=prompt
)
return response.text |