Update app.py

app.py

@@ -6,15 +6,16 @@ from io import BytesIO
 from tqdm import tqdm
 import time
 
-# Definindo as informações do repositório e a trigger word
+# Setting the repository information and the trigger word
 repo = "artificialguybr/StudioGhibli.Redmond-V2"
 trigger_word = "Studio Ghibli, StdGBRedmAF"
 
 def generate_image(prompt):
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
-    #token = os.getenv("API_TOKEN")  # Descomente e use seu token da API do Hugging Face
+    # Uncomment and use your Hugging Face API token
+    # token = os.getenv("API_TOKEN")
     headers = {
-        #"Authorization": f"Bearer {token}"
+        # "Authorization": f"Bearer {token}"
     }
     full_prompt = f"{prompt} {trigger_word}"
     payload = {
@@ -31,7 +32,7 @@ def generate_image(prompt):
     while True:
         response = requests.post(api_url, headers=headers, json=payload)
         if response.status_code == 200:
-            return Image.open(BytesIO(response.content))  # Alterado para corresponder ao primeiro código
+            return Image.open(BytesIO(response.content))
         elif response.status_code == 503:
             time.sleep(1)
             pbar.update(1)
@@ -41,12 +42,19 @@ def generate_image(prompt):
         else:
             raise Exception(f"API Error: {response.status_code}")
 
+css = """
+body { font-family: 'Helvetica Neue', Helvetica, Arial, sans-serif; }
+h1 { color: #333; }
+.gradio-container { max-width: 700px; margin: auto; }
+"""
+
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(lines=2, placeholder="Digite seu prompt aqui..."),
+    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
     outputs="image",
-    title="Gerador de Imagens Studio Ghibli V2",
-    description="Insira um prompt de texto para gerar uma imagem com o estilo do Studio Ghibli."
+    title="Studio Ghibli Image Generator V2",
+    description="Enter a text prompt to generate an image in the style of Studio Ghibli.",
+    css=css
 )
 
-iface.launch()
+iface.launch()