Docfile's picture
Update app.py
2ec3a11 verified
raw
history blame
3.19 kB
import gradio as gr
import google.generativeai as genai
import os
token=os.environ.get("TOKEN")
e = ""
genai.configure(
api_key=token
)
generation_config = {
"temperature": 1,
"top_p": 0.95,
"top_k": 64,
"max_output_tokens": 8192,
}
safety_settings = [
{
"category": "HARM_CATEGORY_HARASSMENT",
"threshold": "BLOCK_NONE"
},
{
"category": "HARM_CATEGORY_HATE_SPEECH",
"threshold": "BLOCK_NONE"
},
{
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
"threshold": "BLOCK_NONE"
},
{
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
"threshold": "BLOCK_NONE"
},
]
mm = """ resous cet exercice. tu répondras en détaillant au maximum ton procédé de calcul. réponse attendue uniquement en Latex
"""
model = genai.GenerativeModel(model_name="gemini-1.5-pro",
generation_config=generation_config ,
safety_settings=safety_settings)
# Fonction pour générer le contenu
def generate_content(image):
global e
if not image:
e =" djo"
else:
response = model.generate_content([mm,image])
print(response.text)
e= response.text
return e
markdown = r"""
{e}
""".format(e=e)
iface = gr.Interface(fn=generate_content, inputs=gr.Image(type='pil'),outputs= gr.Markdown(markdown, latex_delimiters=[{ "left":"$$", "right":"$$", "display": True },{ "left":"$", "right":"$", "display": True },
{
"left": "\\(",
"right": "\\)",
"display": True
}, {
"left": "\\begin\{equation\}",
"right": "\\end\{equation\}",
"display": True
}, {
"left": "\\begin\{align\}",
"right": "\\end\{align\}",
"display": True
}, {
"left": "\\begin\{alignat\}",
"right": "\\end\{alignat\}",
"display": True
}, {
"left": "\\begin\{gather\}",
"right": "\\end\{gather\}",
"display": True
}, {
"left": "\\begin\{CD\}",
"right": "\\end\{CD\}",
"display": True
}, {
"left": "\\[",
"right": "\\]",
"display": True
} ]))
iface.launch()