Update app.py

app.py

@@ -105,7 +105,7 @@ def analyze_dna(sequence, password):
         fig, ax = plt.subplots(figsize=(10, 6))
         ax.barh(top_5_labels, top_5_probs, color='skyblue')
         ax.set_xlabel('Probability')
-        ax.set_title('Top 5 Most Likely Labels')
+        ax.set_title('Top 5 Most Likely Nations of Origin')
         plt.gca().invert_yaxis()
 
         buf = io.BytesIO()
@@ -114,7 +114,7 @@ def analyze_dna(sequence, password):
         image_base64 = base64.b64encode(buf.read()).decode('utf-8')
         buf.close()
 
-        return f'<img src="data:image/png;base64,{image_base64}" />'
+        return result, f'<img src="data:image/png;base64,{image_base64}" />'
 
     except Exception as e:
         return {"error": str(e)}, ""
@@ -123,7 +123,7 @@ def analyze_dna(sequence, password):
 demo = gr.Interface(
     fn=analyze_dna,
     inputs=[gr.Textbox(label="DNA Sequence"), gr.Textbox(label="Password", type="password")],
-    outputs=["html"]
+    outputs=["json", "html"]
 )
 
 # Launch the interface