CommonLit / app.py
sifaaral's picture
Upload 3 files
a80d3f7 verified
raw
history blame
1.03 kB
import streamlit as st
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
# Model ve tokenizer'ı yükle
model_path = 'microsoft/deberta-xlarge'
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForSequenceClassification.from_pretrained(model_path)
# Streamlit uygulaması
st.title('DeBERTa-XLarge Model ile Metin Sınıflandırma')
# Kullanıcıdan metin girişi al
user_input = st.text_area("Metni Buraya Yazın:", height=200)
if st.button("Tahmin Et"):
if user_input:
# Tokenizasyon
inputs = tokenizer(user_input, return_tensors='pt', padding=True, truncation=True)
# Modeli çalıştır
with torch.no_grad():
outputs = model(**inputs)
# Tahmini elde et
predictions = torch.argmax(outputs.logits, dim=-1)
# Sonucu göster
st.success(f'Tahmin Sonucu: {predictions.item()}')
else:
st.warning("Lütfen bir metin giriniz.")