Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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 |
-
|
130 |
-
|
131 |
-
|
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 = [
|