Update app.py

app.py

@@ -49,15 +49,21 @@ final_answer = FinalAnswerTool()
 # custom_role_conversions=None,
 # )
 
+# model = TransformersModel(
+#     model_id="bigscience/bloomz-1b1",
+#     device_map="auto",          # Colab GPU 자동 할당
+#     torch_dtype="auto",         # 메모리 절약
+#     max_new_tokens=512,         # (= max_tokens 별칭으로도 동작)
+#     temperature=0.7,
+# )
+
 model = TransformersModel(
-    model_id="bigscience/bloomz-1b1",
-    device_map="auto",          # Colab GPU 자동 할당
-    torch_dtype="auto",         # 메모리 절약
-    max_new_tokens=512,         # (= max_tokens 별칭으로도 동작)
-    temperature=0.7,
+    model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
+    dtype="auto",
+    max_tokens=1024,
+    temperature=0.5
 )
 
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)