Update app.py

app.py

@@ -49,21 +49,6 @@ def query_image(prompt):
     time.sleep(delay)
   return None
 
-def query_text_generation(prompt):
-  max_retries=5
-  delay=10
-  payload={"inputs":f"give me a short story about {prompt}"}
-  for attempt in range(max_retries):
-    response=requests.post(txt_API,headers=headers,json=payload)
-    if response.status_code == 200:
-      result=response.json()
-      if isinstance(result,list) and len(result)>0:
-        return result[0].get("generated_text","Text Generation Error")
-      elif isinstance(result,dict) and "generated_text" in result:
-        return result["generated_text"]
-    print(f"⚠️ Text generation failed, retrying in {delay} seconds...")
-    time.sleep(delay)
-  return None
   
 
 
@@ -76,7 +61,6 @@ def process_audio(audio_path):
     translation=query_translation(tamil_text)
     translated_text=translation.get("translated_text","Translation error")
     image_path=query_image(translated_text)
-    story=query_text_generation(translated_text)
     return tamil_text,translated_text,image_path,story
   except Exception as e:
     return None,str(e),None
@@ -88,7 +72,7 @@ iface=gr.Interface(
         gr.Textbox(label="Tamil Text"),
         gr.Textbox(label="English Translation"),
         gr.Image(label="Generated Image")
-        gr.Textbox(label="Story")
+        
     ],
     title="Speech-to-Image Generation"
 )