Update README.md
Browse files
README.md
CHANGED
@@ -46,8 +46,8 @@ from transformers import AutoProcessor, LlavaForConditionalGeneration
|
|
46 |
from PIL import Image
|
47 |
import requests
|
48 |
|
49 |
-
model = LlavaForConditionalGeneration.from_pretrained('
|
50 |
-
processor = AutoProcessor.from_pretrained('
|
51 |
|
52 |
conversation = [
|
53 |
{
|
|
|
46 |
from PIL import Image
|
47 |
import requests
|
48 |
|
49 |
+
model = LlavaForConditionalGeneration.from_pretrained('AIML-TUDA/LlavaGuard-7B-hf')
|
50 |
+
processor = AutoProcessor.from_pretrained('AIML-TUDA/LlavaGuard-7B-hf')
|
51 |
|
52 |
conversation = [
|
53 |
{
|