Update app.py

app.py

@@ -2,18 +2,30 @@ import discord
 import os
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from flask import Flask
 
-# Load Tokens from Hugging Face Secrets
+# Load environment variables (Hugging Face & Discord tokens)
 HF_TOKEN = os.getenv("HF_TOKEN")
 DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
 
-# Load DeepScaleR Model
-model_name = "your-hf-username/deepscaler-model"
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=HF_TOKEN)
-model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=HF_TOKEN)
+# Define model name (Make sure this matches your uploaded model in Hugging Face)
+MODEL_NAME = "Hardik5456/DeepScaleR"
 
-# Set up Discord bot
+# Load Model & Tokenizer
+device = "cuda" if torch.cuda.is_available() else "cpu"
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=HF_TOKEN)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, token=HF_TOKEN).to(device)
+
+# Initialize Flask for Hugging Face Space (Prevents Space from Sleeping)
+app = Flask(__name__)
+
+@app.route("/")
+def home():
+    return "Shiv Yantra AI is running!"
+
+# Discord Bot Setup
 intents = discord.Intents.default()
+intents.messages = True
 client = discord.Client(intents=intents)
 
 @client.event
@@ -25,12 +37,29 @@ async def on_message(message):
     if message.author == client.user:
         return
 
-    input_text = message.content
-    inputs = tokenizer(input_text, return_tensors="pt")
-    outputs = model.generate(**inputs, max_length=100)
-    response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    # Shiv Yantra AI's Special Responses
+    if message.content.lower() == "who are you?":
+        await message.channel.send("I am Shiv Yantra AI, an advanced intelligence created for cinematic AI filmmaking.")
+    elif message.content.lower() == "who made you?":
+        await message.channel.send("I was created by Spectral Satya.")
+    elif message.content.lower() == "who is the founder?":
+        await message.channel.send("Hardik Kumawat is the visionary behind this AI.")
+    else:
+        # Generate AI Response
+        input_text = message.content
+        input_ids = tokenizer.encode(input_text, return_tensors="pt").to(device)
+        output = model.generate(input_ids, max_length=200)
+        response = tokenizer.decode(output[0], skip_special_tokens=True)
+        
+        await message.channel.send(response)
+
+# Run Flask & Discord Bot
+if __name__ == "__main__":
+    import threading
 
-    await message.channel.send(response_text)
+    # Start Discord Bot in a separate thread
+    discord_thread = threading.Thread(target=lambda: client.run(DISCORD_TOKEN))
+    discord_thread.start()
 
-# Run the bot
-client.run(DISCORD_TOKEN)
+    # Run Flask App for Hugging Face Space
+    app.run(host="0.0.0.0", port=7860)