Update app.py

app.py

@@ -1,27 +1,36 @@
-import streamlit as st
-import transformers
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+import requests
+import json
 
-#from transformers import T5Tokenizer, T5ForConditionalGeneration
-#model_name = "t5-base"
-#tokenizer = T5Tokenizer.from_pretrained(model_name)
-#model = T5ForConditionalGeneration.from_pretrained(model_name)
-model_name = "indonesia/gpt-2-small-indonesia"
-model = GPT2LMHeadModel.from_pretrained(model_name)
-tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+# Replace with your actual API key
+API_KEY = "sk-3df8d86d5a4b49cb9240077792c55d0a"
+API_URL = "https://api.deepseek.com/v1/chat/completions"  # Example endpoint, check documentation
 
-def generate_response(input_text):
-    input_ids = tokenizer.encode(input_text, return_tensors='pt')
-    outputs = model.generate(input_ids,
-                             min_length=5,
-                             max_length=300,
-                             do_sample=True, num_beams=5, no_repeat_ngram_size=2)
-    generated_text = tokenizer.decode(
-        outputs[0], skip_special_tokens=True)
-    return generated_text
+# Define the prompt in Bahasa Indonesia
+prompt = "Halo, bisakah kamu membantu saya membuat rencana perjalanan ke Bali?"
 
+# Prepare the payload
+payload = {
+    "model": "deepseek-v3",  # Replace with the model you want to use
+    "messages": [
+        {"role": "user", "content": prompt}
+    ],
+    "max_tokens": 150,  # Adjust as needed
+    "temperature": 0.7,  # Adjust for creativity
+}
 
-prompt = st.chat_input(placeholder="Say Something!",key=None, max_chars=None, disabled=False, on_submit=None, args=None, kwargs=None)
-if prompt:
-   with st.chat_message(name="AI",avatar=None):
-    st.write(generate_response(prompt))
+# Set headers
+headers = {
+    "Authorization": f"Bearer {API_KEY}",
+    "Content-Type": "application/json",
+}
+
+# Make the API request
+response = requests.post(API_URL, headers=headers, data=json.dumps(payload))
+
+# Check the response
+if response.status_code == 200:
+    result = response.json()
+    generated_text = result['choices'][0]['message']['content']
+    print("Generated Text:", generated_text)
+else:
+    print("Error:", response.status_code, response.text)