|
|
import gradio as gr |
|
|
import os |
|
|
os.environ["KERAS_BACKEND"] = "tensorflow" |
|
|
import keras |
|
|
import keras_nlp |
|
|
import utils |
|
|
|
|
|
from deep_translator import GoogleTranslator |
|
|
|
|
|
css = """ |
|
|
html, body { |
|
|
margin: 0; |
|
|
padding: 0; |
|
|
height: 100%; |
|
|
overflow: hidden; |
|
|
} |
|
|
body::before { |
|
|
content: ''; |
|
|
position: fixed; |
|
|
top: 0; |
|
|
left: 0; |
|
|
width: 100vw; |
|
|
height: 100vh; |
|
|
background-image: url('https://stsci-opo.org/STScI-01J5E849R5W27ZZ2C3QAE9ET75.png'); |
|
|
background-size: cover; |
|
|
background-repeat: no-repeat; |
|
|
opacity: 0.35; |
|
|
background-position: center; |
|
|
z-index: -1; |
|
|
} |
|
|
.gradio-container { |
|
|
display: flex; |
|
|
justify-content: center; |
|
|
align-items: center; |
|
|
height: 100vh; |
|
|
} |
|
|
""" |
|
|
|
|
|
gemma_lm = keras_nlp.models.CausalLM.from_preset("hf://sultan-hassan/CosmoGemma_2b_en") |
|
|
chat = utils.ChatState(gemma_lm) |
|
|
|
|
|
def launch(message): |
|
|
try: |
|
|
|
|
|
message_en = GoogleTranslator(source='auto', target='en').translate(message) |
|
|
|
|
|
|
|
|
response_en = chat.send_message(message_en) |
|
|
|
|
|
|
|
|
response_kk = GoogleTranslator(source='en', target='kk').translate(response_en) |
|
|
|
|
|
return response_kk |
|
|
except Exception as e: |
|
|
return f"Аударма немесе модельде қате орын алды: {str(e)}" |
|
|
|
|
|
examples = [ |
|
|
'Космологиядағы негізгі зерттеу әдістері қандай?', |
|
|
'Инфляциялық модельдерде қандай байланыстар инфляция параметрлерін болжауға әсер етеді?', |
|
|
'Астрономиядағы сигналдағы артық қуаттың себебі неде болуы мүмкін?', |
|
|
] |
|
|
|
|
|
iface = gr.Interface(launch, |
|
|
inputs="text", |
|
|
outputs="text", |
|
|
examples=examples, |
|
|
css=css, |
|
|
title="CosmoGemma қазақша сөйлейді 👋", |
|
|
description="CosmoGemma моделін қазақша сұрақтармен пайдаланыңыз, жауаптарды қазақша алыңыз.") |
|
|
|
|
|
iface.launch() |
|
|
|
