easyreadDemo / src /services /gemini.py
system's picture
system HF Staff
Deploy 53c0813 from hugging_face
1c57e18 verified
Raw History Blame Contribute Delete
311 Bytes
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