sa / app.py
Sameh891's picture
Create app.py
965eda8 verified
raw
history blame contribute delete
865 Bytes
import gradio as gr
from transformers import pipeline
from PIL import Image
# استدعاء النموذج مباشرة من Hugging Face
model_id = "microsoft/llava-med-v1.5-mistral-7b"
# إعداد خط المعالجة (Pipeline)
# ملاحظة: سيتم تحميل النموذج تلقائياً عند التشغيل الأول
pipe = pipeline("image-to-text", model=model_id)
def analyze_radiology(image):
# إرسال الصورة للنموذج وطلب التحليل
result = pipe(image, prompt="USER: <image>\nDescribe this medical image and find any abnormalities. ASSISTANT:")
return result[0]['generated_text']
# واجهة مستخدم بسيطة ستعمل كـ API لموقعك
demo = gr.Interface(
fn=analyze_radiology,
inputs=gr.Image(type="pil"),
outputs="text",
title="Eldaly Rehab Scholar AI"
)
demo.launch()