felfri commited on
Commit
fed5d71
1 Parent(s): e1e85bd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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('LukasHug/LlavaGuard-7B-hf')
50
- processor = AutoProcessor.from_pretrained('LukasHug/LlavaGuard-7B-hf')
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
  {