Spaces:
Runtime error
Runtime error
File size: 1,225 Bytes
e7a20c4 1bae62b 3218808 472e3e3 1bae62b 1543f62 fc050a2 1bae62b e7a20c4 c044ee6 e7a20c4 5c83f2f e7a20c4 58719ad 20a0181 94587dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import gradio as gr
import google.generativeai as genai
import numpy as np
import base64
import pathlib
import textwrap
import os
secret_key =os.getenv("API_KEY")
# secret_key = userdata.get('API_KEY')
genai.configure(api_key = secret_key)
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", sources=['upload', 'webcam'], type="pil"),
outputs=gr.Textbox(label="Your prescription is here:"),
title="Medicine Prescription",
description="Find Prescription to any medicine",
allow_flagging="never",
examples = [
["examples/Combiflam.jpg"],
["examples/Zinetac.jpg"]
])
if __name__ == "__main__":
interface.launch() |