updated

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from transformers import pipeline
+
+# Initialize the pipeline
+pipe = pipeline("feature-extraction", model="ECOFRI/CXR-LLAVA-v2", trust_remote_code=True)
+
+# Define a function to process inputs and return outputs
+def extract_features(text):
+    features = pipe(text)
+    return features
+
+# Create a Gradio interface
+iface = gr.Interface(
+    fn=extract_features, 
+    inputs="text", 
+    outputs="json",
+    title="Feature Extraction Demo",
+    description="Enter text to extract features using the ECOFRI/CXR-LLAVA-v2 model."
+)
+
+# Launch the interface
+iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+transformers
+gradio
+torch
+einops