ClaudiaRichard commited on
Commit
44e69d2
·
verified ·
1 Parent(s): eadf3ca

update streamlit.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -2
src/streamlit_app.py CHANGED
@@ -1,8 +1,7 @@
1
  import torch
2
  import streamlit as st
3
  from PIL import Image
4
- from transformers import AutoProcessor, BitsAndBytesConfig
5
- from transformers.models.llama.modeling_llama import LlamaForCausalLM as MllamaForConditionalGeneration
6
  import os
7
 
8
  # --- Setup HF cache to writable directory (for Spaces) ---
 
1
  import torch
2
  import streamlit as st
3
  from PIL import Image
4
+ from transformers import MllamaForConditionalGeneration, AutoProcessor, BitsAndBytesConfig
 
5
  import os
6
 
7
  # --- Setup HF cache to writable directory (for Spaces) ---