Upload 3 files

README.md

@@ -1,12 +1,13 @@
 ---
-title: AdViduCopy
-emoji: 💻
-colorFrom: purple
-colorTo: green
+title: AdLLaVA
+emoji: 📊
+colorFrom: yellow
+colorTo: purple
 sdk: gradio
 sdk_version: 4.23.0
 app_file: app.py
 pinned: false
+license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,18 @@
+import gradio as gr
+import torch
+from transformers import BitsAndBytesConfig, pipeline
+
+# Load model directly
+
+
+model_id = "marianna13/llava-phi-2-3b"
+pipe = pipeline("image-to-text", model=model_id)
+
+def generate_text(image):
+    max_new_tokens = 200
+    prompt = "USER: <image>\nWhat are the things I should be cautious about when I visit this place?\nASSISTANT:"
+    outputs = pipe(image, prompt=prompt, generate_kwargs={"max_new_tokens": 200})
+    return outputs[0]["generated_text"]
+
+iface = gr.Interface(fn=generate_text, inputs=gr.inputs.Image(), outputs="text")
+iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+transformers==4.37.2
+bitsandbytes==0.41.3
+accelerate==0.25.0
+gradio