Update app.py

app.py

@@ -1,11 +1,9 @@
 import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM
-import streamlit as st
+from transformers import AutoModel
 
-# Загрузка модели и токенизатора
 model_name = "Richieburundi/Ariginalmodel"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model = AutoModel.from_pretrained(model_name, config=None, tokenizer=None)
+import streamlit as st
 
 def generate_text(input_text, max_length=50):
     inputs = tokenizer(input_text, return_tensors="pt")