Aashi's picture
Update app.py
f977c9a
raw
history blame
873 Bytes
import gradio as gr
import google.generativeai as genai
model = genai.GenerativeModel('gemini-pro-vision')
def generate_prescription(image):
response = model.generate_content(["Write a prescription in pointer format ordered by name of medicine, symptoms, primary diagnosis, usage and dosage of medicine in the image. Make sure to ask person to visit doctor if problem presists.", image])
return response.text
interface = gr.Interface(fn=generate_prescription,
inputs=gr.Image(label="Upload image", type="pil"),
outputs=gr.Textbox(label="Your prescription is here:"),
title="Medicine Prescription",
description="Find Prescription to any medicine",
allow_flagging="never",
examples=["Combiflam.jpg", "Zintac.jpg"])
interface.launch(debug = True)