Update app.py

app.py

@@ -1,15 +1,15 @@
 print("start to run")
 import streamlit as st
 import os
-os.system("pip install torch transformers sentencepiece accelerate")
+os.system("pip install torch transformers sentencepiece accelerate torch.utils torchvision")
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-
+print("[code] All module has imported.")
 # モデルとトークナイザの初期化
 
 model = AutoModelForCausalLM.from_pretrained("cyberagent/open-calm-1b", device_map="auto", torch_dtype=torch.float16)
 tokenizer = AutoTokenizer.from_pretrained("cyberagent/open-calm-1b")
-
+print("[code] model loaded")
 # 推論用の関数
 def generate_text(input_text, max_new_tokens, temperature, top_p, repetition_penalty):
     inputs = tokenizer(input_text, return_tensors="pt").to(model.device)