Upload folder using huggingface_hub

web_demo.py

@@ -1,4 +1,5 @@
 from transformers import AutoModel, AutoTokenizer,AutoModelForCausalLM
+import torch
 
 import gradio as gr
 import mdtex2html

web_demo2.py

@@ -1,6 +1,7 @@
 from transformers import AutoModel, AutoTokenizer,AutoModelForCausalLM
 import streamlit as st
 from streamlit_chat import message
+import torch
 
 
 st.set_page_config(

web_demo_old.py

@@ -1,5 +1,7 @@
 from transformers import AutoModel, AutoTokenizer,AutoModelForCausalLM
 import gradio as gr
+import torch
+
 
 # tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
 # model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()

web_demo_vision.py

@@ -1,6 +1,7 @@
 from transformers import AutoModel, AutoTokenizer,AutoModelForCausalLM
 import gradio as gr
 import mdtex2html
+import torch
 
 # tokenizer = AutoTokenizer.from_pretrained("THUDM/visualglm-6b", trust_remote_code=True)
 # model = AutoModel.from_pretrained("THUDM/visualglm-6b", trust_remote_code=True).half().cuda()