Update app.py

app.py

@@ -21,10 +21,10 @@ def translate_text(tamil_text):
     translation = translation_tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
     return translation
 
-def query_gpt_neo(translated_text, max_words):
+def query_gpt_neo(translated_text):
     prompt = f"Continue the story based on the following text: {translated_text}"
     inputs = gpt_tokenizer(prompt, return_tensors="pt")
-    outputs = gpt_model.generate(inputs['input_ids'], max_length=max_words, num_return_sequences=1)
+    outputs = gpt_model.generate(inputs['input_ids'], max_length=100, num_return_sequences=1)
     creative_text = gpt_tokenizer.decode(outputs[0], skip_special_tokens=True)
     return creative_text
 
@@ -42,13 +42,13 @@ def query_image(payload):
     else:
         return f"Error: {response.status_code} - {response.text}"
 
-def process_input(tamil_input, max_words):
+def process_input(tamil_input):
     try:
         # Translate the input text
         translated_output = translate_text(tamil_input)
         
         # Generate creative text using GPT-Neo
-        creative_output = query_gpt_neo(translated_output, max_words)
+        creative_output = query_gpt_neo(translated_output)
         
         # Generate an image using Hugging Face's FLUX model
         image_bytes = query_image({"inputs": translated_output})
@@ -58,22 +58,17 @@ def process_input(tamil_input, max_words):
     except Exception as e:
         return f"Error occurred: {str(e)}", "", None
 
-# Create a Gradio interface with interactive elements
+# Create a Gradio interface
 interface = gr.Interface(
     fn=process_input,
-    inputs=[
-        gr.Textbox(label="Input Tamil Text", placeholder="Enter your Tamil text here..."),
-        gr.Slider(label="Max Words for Creative Text", minimum=50, maximum=200, step=10, value=100)
-    ],
+    inputs=[gr.Textbox(label="Input Tamil Text")],
     outputs=[
         gr.Textbox(label="Translated Text"),
         gr.Textbox(label="Creative Text"),
         gr.Image(label="Generated Image")
     ],
-    title="TRANSART - Multimodal AI App",
-    description="Enter Tamil text to translate to English, generate creative text, and produce an image based on the translated text.",
-    theme="compact",  # Use the 'compact' theme for a cleaner app look
-    layout="vertical"  # Arrange components vertically for better readability
+    title="TRANSART",
+    description="Enter Tamil text to translate to English and generate an image based on the translated text."
 )
 
 interface.launch()