Dax451 commited on
Commit
678476f
·
verified ·
1 Parent(s): d4904b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -10
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import requests
4
  from PIL import Image
5
  from io import BytesIO
6
- import google.generativeai as genai
7
  from dotenv import load_dotenv
8
  import time
9
  import base64
@@ -126,16 +126,17 @@ class GeminiPromptGenerator:
126
 
127
  def __init__(self):
128
  """Inizializza il generatore di prompt"""
129
- genai.configure(api_key=os.environ.get("GEMINI_API_KEY"))
130
- self.model = genai.GenerativeModel(
131
- model_name="gemini-2.0-pro-exp-02-05",
132
- generation_config=genai.types.GenerationConfig(
133
- temperature=1.4, # Aumentata per più creatività
134
- top_p=0.99, # Aumentato per più variabilità
135
- top_k=40, # Aumentato per considerare più opzioni
136
- max_output_tokens=8192
137
- )
138
  )
 
 
 
 
 
 
 
139
 
140
  # Lista di stili artistici per variare i prompt
141
  self.art_styles = [
 
3
  import requests
4
  from PIL import Image
5
  from io import BytesIO
6
+ from openai import OpenAI
7
  from dotenv import load_dotenv
8
  import time
9
  import base64
 
126
 
127
  def __init__(self):
128
  """Inizializza il generatore di prompt"""
129
+ self.client = OpenAI(
130
+ base_url="https://openrouter.ai/api/v1",
131
+ api_key=os.environ.get("OPENROUTER_API_KEY")
 
 
 
 
 
 
132
  )
133
+ self.model = {
134
+ "model": "google/gemma-7b:free",
135
+ "temperature": 1.4, # Aumentata per più creatività
136
+ "top_p": 0.99, # Aumentato per più variabilità
137
+ "top_k": 40, # Aumentato per considerare più opzioni
138
+ "max_tokens": 8192 # Equivalente a max_output_tokens
139
+ }
140
 
141
  # Lista di stili artistici per variare i prompt
142
  self.art_styles = [