Update app.py

app.py

@@ -3,7 +3,8 @@ from flask_socketio import SocketIO, emit
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import numpy as np
-from PIL import Image, ImageDraw, io
+from PIL import Image, ImageDraw
+import io  # Changed this line - io is a built-in Python module
 import time
 import threading
 import random
@@ -12,7 +13,7 @@ app = Flask(__name__)
 socketio = SocketIO(app)
 
 # Initialize model with lower precision
-MODEL_NAME = "Qwen/Qwen-1_5B-Chat"
+MODEL_NAME = "Qwen/Qwen-1_8B-Chat"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,