Update app.py

app.py

@@ -1,17 +1,23 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
 
-# بارگذاری مدل از Hugging Face
+# بارگذاری مدل "HooshvareLab/bert-fa-base-uncased" از Hugging Face
 @st.cache_resource
 def load_model():
-    generator = pipeline("text-generation", model="gpt2")  # یا هر مدل مناسب دیگر
-    return generator
+    tokenizer = AutoTokenizer.from_pretrained("HooshvareLab/bert-fa-base-uncased")
+    model = AutoModelForSequenceClassification.from_pretrained("HooshvareLab/bert-fa-base-uncased")
+    return tokenizer, model
 
-model = load_model()
+tokenizer, model = load_model()
 
-def generate_response(prompt):
-    response = model(prompt, max_length=100, num_return_sequences=1)  # تنظیم max_length برای پاسخ‌های طولانی‌تر
-    return response[0]["generated_text"]
+# تابع برای تولید پاسخ (فعلا فقط متن ورودی را برمی‌گرداند)
+def generate_response(text):
+    inputs = tokenizer(text, return_tensors="pt")
+    outputs = model(**inputs)
+    # TODO: پردازش خروجی مدل و تولید پاسخ مناسب
+    # در حال حاضر، فقط متن ورودی را برمی‌گردانیم
+    response = text
+    return response
 
 def main():
     st.set_page_config(page_title="دستیار هوش مصنوعی", layout="wide")