Update app.py

app.py

@@ -15,14 +15,13 @@ def translate_text(tamil_text):
     translation = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
     return translation
 
-def query_gemini_api(translated_text, max_output_tokens=300):
+def query_gemini_api(translated_text):
     gemini_api_key = os.getenv('GEMINI_API_KEY')
-    url = f"https://generativelanguage.googleapis.com/v1beta2/models/gemini-1.5:generateText?key={gemini_api_key}"
+    url = f"https://generativelanguage.googleapis.com/v1beta2/models/gemini-1.5-flash:generateText?key={gemini_api_key}"
     headers = {"Content-Type": "application/json"}
     payload = {
         "prompt": {"text": translated_text},
-        "temperature": 0.7,
-        "max_output_tokens": max_output_tokens
+        "temperature": 0.7
     }
     
     response = requests.post(url, headers=headers, json=payload)
@@ -41,19 +40,16 @@ def query_image(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.content
 
-def process_input(tamil_input, max_output_tokens):
+def process_input(tamil_input):
     translated_output = translate_text(tamil_input)
-    creative_output = query_gemini_api(translated_output, max_output_tokens)
+    creative_output = query_gemini_api(translated_output)
     image_bytes = query_image({"inputs": translated_output})
     image = Image.open(io.BytesIO(image_bytes))
     return translated_output, creative_output, image
 
 iface = gr.Interface(
     fn=process_input,
-    inputs=[
-        gr.Textbox(label="Input Tamil Text"),
-        gr.Slider(50, 1000, value=300, step=50, label="Max Output Tokens for Creative Text")
-    ],
+    inputs=[gr.Textbox(label="Input Tamil Text")],
     outputs=[
         gr.Textbox(label="Translated Text"),
         gr.Textbox(label="Creative Text"),