Aashi's picture
Update app.py
1bae62b
raw
history blame
884 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=["/content/med1.jpg", "/content/med2.jpg"])
interface.launch(debug = True)