File size: 780 Bytes
8c7ad0f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import os
import google.generativeai as genai
# Fetch the API key from the environment variable (Space secret)
api_key = os.getenv("GOOGLE_API_KEY")
if not api_key:
raise ValueError("GOOGLE_API_KEY secret not set in Hugging Face Space settings.")
# Configure the Gemini API
genai.configure(api_key=api_key)
# Initialize the model
# You can change 'gemini-1.5-flash-latest' to 'gemini-1.5-pro-latest' or other models
model = genai.GenerativeModel('gemini-1.5-flash-latest')
# Get user input from the terminal
prompt = input("Enter your prompt for Gemini: ")
print("\nGenerating response...")
# Generate content
response = model.generate_content(prompt)
# Print the response text
print("\n--- Gemini Response ---")
print(response.text)
print("-----------------------\n") |