Spaces:

JoannaKOKO
/

Tarot_Explainer_gpu

Sleeping

JoannaKOKO commited on Mar 25

Commit

148b365

verified ·

1 Parent(s): af9fb37

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import AutoProcessor, AutoModelForImageTextToText
 import torch
 import spaces
 from peft import PeftModel
-from transformers import AutoModelForCausalLM
 # Function to process vision information
 def process_vision_info(messages: list[dict]) -> list[Image.Image]:

 import torch
 import spaces
 from peft import PeftModel
+from transformers import AutoModelForCausalLM, AutoTokenizer
 # Function to process vision information
 def process_vision_info(messages: list[dict]) -> list[Image.Image]: