Update main.py

main.py

@@ -6,6 +6,7 @@ from peft import PeftModel, get_peft_config
 import json
 import torch
 
+'''
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 
@@ -32,6 +33,7 @@ model = base_model.to(device)
 
 # 加载 tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+'''
 
 def format_instruction(task, text):
     string = f"""### 指令:
@@ -107,17 +109,17 @@ async def say_hello():
 # POST请求接口
 @app.post("/process")
 async def process_text(request: ProcessRequest):
-    if request.method == 0:
+    if request.method == "0":
         #processed_text = request.text.upper()
         processed_text = "predict(request.text, 0)"
-    elif request.method == 1:
+    elif request.method == "1":
         #processed_text = request.text.lower()
         processed_text = "predict(request.text, 1)"
-    elif request.method == 2:
+    elif request.method == "2":
         processed_text = "request.text[::-1]"  # 反转字符串
     else:
         processed_text = "request.text"
 
     return {"original_text": request.text, "processed_text": processed_text, "method": request.method}
 
-print("fastapi done")
+print("fastapi done 1")