dialog

app.py

@@ -0,0 +1,86 @@
+import gradio as gr
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+# Initialize model and tokenizer
+model = None
+tokenizer = None
+
+print("🚀 Initializing DialoGPT-medium model...")
+
+try:
+    print("📥 Loading DialoGPT-medium model...")
+    model_name = "microsoft/DialoGPT-medium"
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    print("✅ DialoGPT-medium model loaded successfully!")
+    
+    # Add padding token if it doesn't exist
+    if tokenizer.pad_token is None:
+        tokenizer.pad_token = tokenizer.eos_token
+    
+except Exception as e:
+    print(f"❌ Failed to load DialoGPT-medium model: {e}")
+    model = None
+    tokenizer = None
+
+def respond(message, chat_history):
+    """Respond to user message using DialoGPT"""
+    if model is None or tokenizer is None:
+        return "Model not loaded. Please try again later."
+    
+    # Build conversation history
+    conversation = ""
+    for turn in chat_history:
+        conversation += f"User: {turn[0]}\nBot: {turn[1]}\n"
+    
+    conversation += f"User: {message}\nBot:"
+    
+    # Encode and generate
+    inputs = tokenizer.encode(conversation, return_tensors='pt', max_length=1024, truncation=True)
+    
+    with torch.no_grad():
+        outputs = model.generate(
+            inputs,
+            max_length=len(inputs[0]) + 128,
+            pad_token_id=tokenizer.eos_token_id,
+            do_sample=True,
+            temperature=0.7,
+            top_k=50,
+            top_p=0.95,
+            repetition_penalty=1.2
+        )
+    
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    response = response.split("Bot:")[-1].strip()
+    
+    # Clean response
+    if "\nUser:" in response:
+        response = response.split("\nUser:")[0]
+    
+    chat_history.append((message, response))
+    return "", chat_history
+
+# Create the chat interface
+demo = gr.ChatInterface(
+    fn=respond,
+    title="💬 GihonTech AI Conversation Assistant",
+    description="Chat with an AI powered by Microsoft's DialoGPT-medium model",
+    examples=[
+        "Hello! How are you today?",
+        "What can you help me with?",
+        "Tell me about artificial intelligence",
+        "What's your favorite programming language?",
+    ],
+    cache_examples=False,
+    retry_btn=None,
+    undo_btn="↩️ Undo",
+    clear_btn="🗑️ Clear"
+)
+
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )

requirements.txt

@@ -0,0 +1,4 @@
+torch>=2.0.0
+transformers>=4.30.0
+gradio>=4.0.0
+accelerate>=0.20.0