rinrikatoki commited on
Commit
be1a16c
·
verified ·
1 Parent(s): 4d73c77

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -61,3 +61,12 @@ tokenizer = AutoTokenizer.from_pretrained(
61
  )
62
 
63
  print("✅ مدل و توکنایزر با موفقیت بارگذاری شدند.")
 
 
 
 
 
 
 
 
 
 
61
  )
62
 
63
  print("✅ مدل و توکنایزر با موفقیت بارگذاری شدند.")
64
+
65
+ while True:
66
+ prompt = input("📝 یک دستور وارد کن (exit برای خروج): ")
67
+ if prompt.lower() == "exit":
68
+ break
69
+ inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
70
+ outputs = model.generate(**inputs, max_new_tokens=200)
71
+ response = tokenizer.decode(outputs[0], skip_special_tokens=True)
72
+ print("🧠 پاسخ:", response)